From b4cb8edb51a87301fcb70039fb550fc6aca730ac Mon Sep 17 00:00:00 2001 From: Haoyu Chen <65992192+haoyuc@users.noreply.github.com> Date: Sat, 22 Jul 2023 23:01:59 +0800 Subject: [PATCH] Update main_train_psnr.py --- main_train_psnr.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/main_train_psnr.py b/main_train_psnr.py index d131bbb..24e9de5 100644 --- a/main_train_psnr.py +++ b/main_train_psnr.py @@ -326,6 +326,8 @@ def main(json_path='options/train_msrresnet_psnr.json'): writer.add_scalar('PSNRY', avg_psnrY, global_step=current_step) writer.add_scalar('SSIMY', avg_ssimY, global_step=current_step) + writer.add_scalar('PSNR', avg_psnr, global_step=current_step) + writer.add_scalar('SSIM', avg_ssim, global_step=current_step) writer.add_scalar('LPIPS', avg_lpips, global_step=current_step)