diff --git a/video_prediction_savp/HPC_scripts/train_era5_template.sh b/video_prediction_savp/HPC_scripts/train_era5_template.sh index f368a8a1a506f4b6127e01b3f16df5f4d648241c..5342c32a8883c6c0e1534b5a05ef9e59d9c92ffa 100644 --- a/video_prediction_savp/HPC_scripts/train_era5_template.sh +++ b/video_prediction_savp/HPC_scripts/train_era5_template.sh @@ -34,18 +34,16 @@ if [ -z ${VIRTUAL_ENV} ]; then fi fi - - - # declare directory-variables which will be modified appropriately during Preprocessing (invoked by mpi_split_data_multi_years.py) source_dir=/p/project/deepacf/deeprain/video_prediction_shared_folder/preprocessedData/ destination_dir=/p/project/deepacf/deeprain/video_prediction_shared_folder/models/ -# for choosing the model for choosing the model, convLSTM,savp, mcnet,vae +# 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"/" -# rund training -srun python ../scripts/train_dummy.py --input_dir ${source_dir}/tfrecords/ --dataset era5 --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir}/${model}/ +# run training +srun python ../scripts/train_dummy.py --input_dir ${source_dir}/tfrecords/ --dataset era5 --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir} diff --git a/video_prediction_savp/Zam347_scripts/train_era5_template.sh b/video_prediction_savp/Zam347_scripts/train_era5_template.sh index d06392621182ea6b046bfda77c93b1d2b422aa56..c8fcf81f717433a10a8dd1e8652a743e5e8b5332 100644 --- a/video_prediction_savp/Zam347_scripts/train_era5_template.sh +++ b/video_prediction_savp/Zam347_scripts/train_era5_template.sh @@ -9,10 +9,11 @@ exit 99 source_dir=/home/${USER}/preprocessedData/ destination_dir=/home/${USER}/models/ -# for choosing the model +# valid identifiers for model-argument are: convLSTM, savp, mcnet and vae model=mcnet model_hparams=../hparams/era5/model_hparams.json +destination_dir=${destination_dir}/${model}/"$(date +"%Y%m%dT%H%M")_"$USER"/" + +# run training +python ../scripts/train_dummy.py --input_dir ${source_dir}/tfrecords/ --dataset era5 --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir} -# execute respective Python-script -python ../scripts/train_dummy.py --input_dir ${source_dir}/tfrecords/ --dataset era5 --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir}/${model}/ -#srun python scripts/train.py --input_dir data/era5 --dataset era5 --model savp --model_hparams_dict hparams/kth/ours_savp/model_hparams.json --output_dir logs/era5/ours_savp