From 783c3d9aa102240b074c48c7c0e78d5144939bd3 Mon Sep 17 00:00:00 2001 From: Michael <m.langguth@fz-juelich.de> Date: Fri, 25 Sep 2020 15:50:11 +0200 Subject: [PATCH] Adaptions and correction to runscript 'train_model_era5_template.sh'. --- .../HPC_scripts/train_model_era5_template.sh | 8 +++++--- 1 file changed, 5 insertions(+), 3 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 f368a8a1..08e1f409 100644 --- a/video_prediction_tools/HPC_scripts/train_model_era5_template.sh +++ b/video_prediction_tools/HPC_scripts/train_model_era5_template.sh @@ -41,11 +41,13 @@ fi 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 ../main_scripts/main_train_models.py --input_dir ${source_dir}/tfrecords/ --dataset era5 --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir}/ -- GitLab