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 e7c6525a197a7acb5cca927cd0280c3628b2c4f9..19af1297cae1869d4d88cf7748fee71c4c4216d4 100644 --- a/video_prediction_tools/HPC_scripts/train_model_era5_template.sh +++ b/video_prediction_tools/HPC_scripts/train_model_era5_template.sh @@ -41,9 +41,9 @@ destination_dir=/p/project/deepacf/deeprain/video_prediction_shared_folder/model # valid identifiers for model-argument are: convLSTM, savp, mcnet and vae model=convLSTM model_hparams=../hparams/era5/${model}/model_hparams.json -destination_dir=${destination_dir}/${model}/"$(date +"%Y%m%dT%H%M")_"$USER"/" +destination_dir_full=${destination_dir}/${model}/"$(date +"%Y%m%dT%H%M")_"$USER"/" # run training -srun python ../main_scripts/main_train_models.py --input_dir ${source_dir}/tfrecords/ --dataset era5 --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir}/ +srun python ../main_scripts/main_train_models.py --input_dir ${source_dir}/tfrecords/ --dataset era5 --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir_full}/