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
Branches
Tags
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 to comment