From 25064c53acc8fce2353c6006aa3140dc9bdba9ee Mon Sep 17 00:00:00 2001 From: Michael <m.langguth@fz-juelich.de> Date: Tue, 9 Feb 2021 20:46:22 +0100 Subject: [PATCH] Further adaptions to the runscripts. --- .../HPC_scripts/preprocess_data_era5_step2_template.sh | 4 +++- .../HPC_scripts/preprocess_data_moving_mnist_template.sh | 2 +- .../HPC_scripts/train_model_era5_template.sh | 4 +++- 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/video_prediction_tools/HPC_scripts/preprocess_data_era5_step2_template.sh b/video_prediction_tools/HPC_scripts/preprocess_data_era5_step2_template.sh index a6b9c79e..85620bfa 100644 --- a/video_prediction_tools/HPC_scripts/preprocess_data_era5_step2_template.sh +++ b/video_prediction_tools/HPC_scripts/preprocess_data_era5_step2_template.sh @@ -43,5 +43,7 @@ sequences_per_file=10 sequence_length=20 # run preprocessing (step 2 where Tf-records are generated) -srun python ../main_scripts/main_preprocess_data_step2.py -input_dir ${source_dir} -output_dir ${destination_dir} -datasplit_config ${datasplit_dir} -hparams_dict_config ${hparams_dict_config} -sequences_per_file ${sequences_per_file} +srun python ../main_scripts/main_preprocess_data_step2.py -input_dir ${source_dir} -output_dir ${destination_dir} + -datasplit_config ${datasplit_dir} -hparams_dict_config ${hparams_dict_config} + -sequences_per_file ${sequences_per_file} diff --git a/video_prediction_tools/HPC_scripts/preprocess_data_moving_mnist_template.sh b/video_prediction_tools/HPC_scripts/preprocess_data_moving_mnist_template.sh index b84ae0a0..fadb8b41 100644 --- a/video_prediction_tools/HPC_scripts/preprocess_data_moving_mnist_template.sh +++ b/video_prediction_tools/HPC_scripts/preprocess_data_moving_mnist_template.sh @@ -37,4 +37,4 @@ source_dir=/p/project/deepacf/deeprain/video_prediction_shared_folder/preprocess destination_dir=/p/project/deepacf/deeprain/video_prediction_shared_folder/preprocessedData/moving_mnist # run Preprocessing (step 2 where Tf-records are generated) -srun python ../video_prediction/datasets/moving_mnist.py ${source_dir} ${destination_dir}/tfrecords +srun python ../video_prediction/datasets/moving_mnist.py ${source_dir} ${destination_dir} 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 f1188834..aa52ce1b 100644 --- a/video_prediction_tools/HPC_scripts/train_model_era5_template.sh +++ b/video_prediction_tools/HPC_scripts/train_model_era5_template.sh @@ -43,9 +43,11 @@ destination_dir=/p/project/deepacf/deeprain/video_prediction_shared_folder/model model=convLSTM datasplit_dict=../data_split/cv_test.json model_hparams=${destination_dir}/model_hparams.json +dataset=era5 # run training -srun python ../main_scripts/main_train_models.py --input_dir ${source_dir} --datasplit_dict ${datasplit_dict} --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} --datasplit_dict ${datasplit_dict} + --dataset ${dataset} --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir} -- GitLab