From 83f2f8d239f4ee0ba14f38c72acbd5510aecf1e2 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 25 Aug 2020 18:11:53 -0600 Subject: [PATCH] more debugging --- codes/models/SRGAN_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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()