Fixed small ytpo in autotune (printing all PSNR)

zhangkaiitugithub · Nov 11, 2021 · b54fe38 · b54fe38
1 parent 7578bec
commit b54fe38
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/opt/autotune.py b/opt/autotune.py
@@ -244,13 +244,13 @@ def recursive_replace(data, variables):
                 if path.isfile(ssim_file_path):
                     with open(ssim_file_path, 'r') as f:
                         ssim = float(f.read())
-                    ssim_txt = f'{psnr:.10f}'
+                    ssim_txt = f'{ssim:.10f}'
                 else:
                     ssim_txt = 'ERR'
                 if path.isfile(lpips_file_path):
                     with open(lpips_file_path, 'r') as f:
                         lpips = float(f.read())
-                    lpips_txt = f'{psnr:.10f}'
+                    lpips_txt = f'{lpips:.10f}'
                 else:
                     lpips_txt = 'ERR'
                 line = f'{path.basename(train_dir.rstrip("/"))}\t{psnr_txt}\t{ssim_txt}\t{lpips_txt}\n'