From 935a7353272f1625f7624317f538aeb7094c34f2 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 25 Aug 2020 17:05:16 -0600 Subject: [PATCH] More dohs --- codes/models/SRGAN_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/SRGAN_model.py b/codes/models/SRGAN_model.py index 47226a52..95832712 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -268,7 +268,7 @@ class SRGANModel(BaseModel): self.get_grad_nopadding = ImageGradientNoPadding().to(self.device) [self.netG, self.netD, self.netD_grad, self.get_grad_nopadding], \ [self.optimizer_G, self.optimizer_D, self.optimizer_D_grad] = \ - amp.initialize([self.netG, self.netD, self.netD_grad, self.get_grad, self.get_grad_nopadding], + amp.initialize([self.netG, self.netD, self.netD_grad, self.get_grad_nopadding], [self.optimizer_G, self.optimizer_D, self.optimizer_D_grad], opt_level=self.amp_level, num_losses=3) else: