From 77d3765364d2e0adb5bcd27bab2f735b6992995f Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 3 Jul 2020 22:20:13 -0600 Subject: [PATCH] Fix new feature loss calc --- codes/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codes/train.py b/codes/train.py index 14de62cd..e429b815 100644 --- a/codes/train.py +++ b/codes/train.py @@ -237,7 +237,7 @@ def main(): visuals = model.get_current_visuals() sr_img = util.tensor2img(visuals['rlt'][b]) # uint8 - gt_img = util.tensor2img(visuals['GT'][b]) # uint8 + #gt_img = util.tensor2img(visuals['GT'][b]) # uint8 # Save SR images for reference img_base_name = '{:s}_{:d}.png'.format(img_name, current_step) @@ -266,7 +266,7 @@ def main(): logger.info('# Validation # PSNR: {:.4e} Fea: {:.4e}'.format(avg_psnr, avg_fea_loss)) # tensorboard logger if opt['use_tb_logger'] and 'debug' not in opt['name']: - tb_logger.add_scalar('val_psnr', avg_psnr, current_step) + #tb_logger.add_scalar('val_psnr', avg_psnr, current_step) tb_logger.add_scalar('val_fea', avg_fea_loss, current_step) #### save models and training states