change monitor to max val psnr

kwea123 · Jan 24, 2021 · 0dd5de2 · 0dd5de2
1 parent 0ceedc7
commit 0dd5de2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -171,8 +171,8 @@ def main(hparams):
     checkpoint_callback = \
         ModelCheckpoint(filepath=os.path.join(f'ckpts/{hparams.exp_name}',
                                                '{epoch:d}'),
-                        monitor='val/loss',
-                        mode='min',
+                        monitor='val/psnr',
+                        mode='max',
                         save_top_k=5)
 
     logger = TestTubeLogger(save_dir="logs",