From f85f1e21dbec8c8bdb830f4bb016b4bc44983abd Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 25 Aug 2020 17:18:52 -0600 Subject: [PATCH] Turns out, can't do that --- codes/models/SRGAN_model.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/codes/models/SRGAN_model.py b/codes/models/SRGAN_model.py index 95832712..9a141ab1 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -291,9 +291,6 @@ class SRGANModel(BaseModel): self.netD_grad = DistributedDataParallel(self.netD_grad, device_ids=[torch.cuda.current_device()], find_unused_parameters=True) - self.get_grad_nopadding = DistributedDataParallel(self.get_grad_nopadding, - device_ids=[torch.cuda.current_device()], - find_unused_parameters=True) else: self.netD = DataParallel(self.netD) if self.spsr_enabled: