Skip to content

Commit

Permalink
Merge pull request #123 from mila-iqia/development
Browse files Browse the repository at this point in the history
Release 3.1.1
  • Loading branch information
mirkobronzi authored Jul 9, 2024
2 parents 03262a2 + 50b17a3 commit fd446d2
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion amlrt_project/train.py
Original file line number Diff line number Diff line change
@@ -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,
)

0 comments on commit fd446d2

Please sign in to comment.