From 21b229d58e21e7308e3697b4414e7e98ad9d3bd0 Mon Sep 17 00:00:00 2001
From: XIANG Yanfei <yanfei.xiang@inria.fr>
Date: Wed, 6 Apr 2022 15:40:20 +0200
Subject: [PATCH] solve the size mismatch issue of the generated *.ckpt files
 based on Luc Giraud(Inria) and Paul Mycek(Cerfacs)'s suggestion

---
 train.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/train.py b/train.py
index f5c6ecf..2f97fbf 100644
--- a/train.py
+++ b/train.py
@@ -77,18 +77,22 @@
     # Create trainer
     logger = TensorBoardLogger("logs", name="helmnet")
 
+        # parser = pl.Trainer.add_argparse_args(parser)
+    args = parser.parse_args()
+    gpu_list = [int(i) for i in args.gpus.split(',')]
+    check_val_every_n_epoch = int(args.check_val_every_n_epoch)
+
     checkpoint_callback = ModelCheckpoint(
         dirpath     = os.getcwd() + "/checkpoints/",
         save_top_k  = 3,
+        every_n_epochs=check_val_every_n_epoch,
         verbose     = True,
         monitor     = "val_loss",
         mode        = "min",
         save_last   = True,
     )
 
-    # parser = pl.Trainer.add_argparse_args(parser)
-    args = parser.parse_args()
-    gpu_list = [int(i) for i in args.gpus.split(',')]
+
 
     # Make trainer
     trainer = pl.Trainer.from_argparse_args(