diff --git a/codes/models/SRGAN_model.py b/codes/models/SRGAN_model.py index aa0fffb2..d1f4b89a 100644 --- a/codes/models/SRGAN_model.py +++ b/codes/models/SRGAN_model.py @@ -77,6 +77,7 @@ class SRGANModel(BaseModel): if self.is_train: self.netD = networks.define_D(opt).to(self.device) if self.spsr_enabled: + logger.info("Defining grad net...") self.netD_grad = networks.define_D(opt, wrap=True).to(self.device) # D_grad if 'network_C' in opt.keys(): @@ -352,7 +353,7 @@ class SRGANModel(BaseModel): self.img_debug_steps = opt['logger']['img_debug_steps'] if 'img_debug_steps' in opt['logger'].keys() else 50 - self.print_network() # print network + #self.print_network() # print network self.load() # load G and D if needed self.load_random_corruptor()