Skip to content
Snippets Groups Projects
Commit 454f746f authored by gong1's avatar gong1
Browse files

Address the env_setup issue for booster

parent 2e873693
Branches
No related tags found
No related merge requests found
Pipeline #45809 failed
#!/bin/bash -x
# 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/era5-Y2010toY2222M01to12-160x128-2970N1500W-T2_MSL_gph500
destination_dir=/p/project/deepacf/deeprain/video_prediction_shared_folder/models/era5-Y2010toY2222M01to12-160x128-2970N1500W-T2_MSL_gph500
# for choosing the model, convLSTM,savp, mcnet,vae
model=convLSTM
model_hparams=/p/project/deepacf/deeprain/bing/ambs/video_prediction_savp/hparams/era5/${model}/model_hparams.json
destination_dir=${destination_dir}/${model}/"$(date +"%Y%m%dT%H%M")_"$USER""
echo $destination_dir
# run training
python3 /p/project/deepacf/deeprain/bing/ambs/video_prediction_savp/scripts/train_dummy.py --input_dir ${source_dir}/tfrecords/ --dataset era5 --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir}/
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment