Merge pull request #123 from mila-iqia/development

Release 3.1.1
mila-iqia · Jul 9, 2024 · fd446d2 · fd446d2
2 parents 03262a2 + 50b17a3
commit fd446d2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/amlrt_project/train.py b/amlrt_project/train.py
@@ -73,6 +73,8 @@ def main():
         output_dir = os.path.join(args.tmp_folder, 'output')
         if not os.path.exists(output_dir):
             os.makedirs(output_dir)
+        if os.path.exists(args.output):
+            rsync_folder(args.output, args.tmp_folder)
     else:
         data_dir = args.data
         output_dir = args.output
@@ -165,7 +167,7 @@ def train_impl(model, datamodule, output, hyper_params, use_progress_bar,
         save_top_k=1,
         verbose=use_progress_bar,
         monitor="val_loss",
-        mode="max",
+        mode="min",
         every_n_epochs=1,
     )