From 829372f7a758cd5323d88bfc47c5467484e20417 Mon Sep 17 00:00:00 2001
From: gong1 <b.gong@fz-juelich.de>
Date: Sat, 22 May 2021 14:40:26 +0200
Subject: [PATCH] add checkpoint to the template
---
.../HPC_scripts/train_model_era5_template.sh | 5 +----
1 file changed, 1 insertion(+), 4 deletions(-)
diff --git a/video_prediction_tools/HPC_scripts/train_model_era5_template.sh b/video_prediction_tools/HPC_scripts/train_model_era5_template.sh
index fff363af..34a4c9dc 100644
--- a/video_prediction_tools/HPC_scripts/train_model_era5_template.sh
+++ b/video_prediction_tools/HPC_scripts/train_model_era5_template.sh
@@ -50,8 +50,5 @@ dataset=era5
# run training
srun python ../main_scripts/main_train_models.py --input_dir ${source_dir} --datasplit_dict ${datasplit_dict} \
- --dataset ${dataset} --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir}
-
-
-
+ --dataset ${dataset} --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir} --checkpoint_dir ${destination_dir}
--
GitLab