fix ddp bug in EDVR

XPixelGroup · Aug 26, 2021 · 54891e9 · 54891e9
1 parent fd260fb
commit 54891e9
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/basicsr/models/edvr_model.py b/basicsr/models/edvr_model.py
@@ -1,5 +1,3 @@
-from torch.nn.parallel import DistributedDataParallel
-
 from basicsr.utils import get_root_logger
 from basicsr.utils.registry import MODEL_REGISTRY
 from .video_base_model import VideoBaseModel
@@ -60,8 +58,5 @@ def optimize_parameters(self, current_iter):
                 logger.warning('Train all the parameters.')
                 for param in self.net_g.parameters():
                     param.requires_grad = True
-                if isinstance(self.net_g, DistributedDataParallel):
-                    logger.warning('Set net_g.find_unused_parameters = False.')
-                    self.net_g.find_unused_parameters = False
 
         super(EDVRModel, self).optimize_parameters(current_iter)