add checkpoint to the template

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}