mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

More dohs

Browse Source
This commit is contained in:
James Betker 2020-08-25 17:05:16 -06:00
parent 53e67bdb9c
commit 935a735327
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/SRGAN_model.py
View File

@ -268,7 +268,7 @@ class SRGANModel(BaseModel):
self.get_grad_nopadding = ImageGradientNoPadding().to(self.device) self.get_grad_nopadding = ImageGradientNoPadding().to(self.device)
[self.netG, self.netD, self.netD_grad, self.get_grad_nopadding], \ [self.netG, self.netD, self.netD_grad, self.get_grad_nopadding], \
[self.optimizer_G, self.optimizer_D, self.optimizer_D_grad] = \ [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], [self.optimizer_G, self.optimizer_D, self.optimizer_D_grad],
opt_level=self.amp_level, num_losses=3) opt_level=self.amp_level, num_losses=3)
else: else: