Skip to content
Snippets Groups Projects
Commit 84a5db9c authored by Michael Langguth's avatar Michael Langguth
Browse files

Adaption of destination_dir according to updated convention for storing...

Adaption of destination_dir according to updated convention for storing trained model (data), i.e. add user and timestamp.
parent 151dcb0c
No related branches found
No related tags found
No related merge requests found
...@@ -34,18 +34,16 @@ if [ -z ${VIRTUAL_ENV} ]; then ...@@ -34,18 +34,16 @@ if [ -z ${VIRTUAL_ENV} ]; then
fi fi
fi fi
# declare directory-variables which will be modified appropriately during Preprocessing (invoked by mpi_split_data_multi_years.py) # 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/ source_dir=/p/project/deepacf/deeprain/video_prediction_shared_folder/preprocessedData/
destination_dir=/p/project/deepacf/deeprain/video_prediction_shared_folder/models/ 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=convLSTM
model_hparams=../hparams/era5/${model}/model_hparams.json model_hparams=../hparams/era5/${model}/model_hparams.json
destination_dir=${destination_dir}/${model}/"$(date +"%Y%m%dT%H%M")_"$USER"/"
# rund training # 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}/${model}/ srun python ../scripts/train_dummy.py --input_dir ${source_dir}/tfrecords/ --dataset era5 --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir}
...@@ -9,10 +9,11 @@ exit 99 ...@@ -9,10 +9,11 @@ exit 99
source_dir=/home/${USER}/preprocessedData/ source_dir=/home/${USER}/preprocessedData/
destination_dir=/home/${USER}/models/ destination_dir=/home/${USER}/models/
# for choosing the model # valid identifiers for model-argument are: convLSTM, savp, mcnet and vae
model=mcnet model=mcnet
model_hparams=../hparams/era5/model_hparams.json 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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment