From be808ba34737beb57d88c5bddc7296f449bc3afa Mon Sep 17 00:00:00 2001 From: gong1 <b.gong@fz-juelich.de> Date: Thu, 3 Sep 2020 10:49:00 +0200 Subject: [PATCH] make train_movingmnist.sh --- .../HPC_scripts/train_movingmnist.sh | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/video_prediction_savp/HPC_scripts/train_movingmnist.sh b/video_prediction_savp/HPC_scripts/train_movingmnist.sh index 6555bb12..b0ec5e83 100755 --- a/video_prediction_savp/HPC_scripts/train_movingmnist.sh +++ b/video_prediction_savp/HPC_scripts/train_movingmnist.sh @@ -4,18 +4,13 @@ #SBATCH --ntasks=1 ##SBATCH --ntasks-per-node=1 #SBATCH --cpus-per-task=1 -<<<<<<< Updated upstream #SBATCH --output=train_moving_mnist-out.%j #SBATCH --error=train_moving_mnist-err.%j -======= -#SBATCH --output=train_era5-out.%j -#SBATCH --error=train_era5-err.%j ->>>>>>> Stashed changes -#SBATCH --time=00:20:00 +#SBATCH --time=23:20:00 #SBATCH --gres=gpu:1 -#SBATCH --partition=develgpus +#SBATCH --partition=gpus #SBATCH --mail-type=ALL -#SBATCH --mail-user=s.stadtler@fz-juelich.de +#SBATCH --mail-user=b.gong@fz-juelich.de ##jutil env activate -p cjjsc42 @@ -48,5 +43,5 @@ model_hparams=../hparams/${dataset}/${model}/model_hparams.json # rund training -srun python ../scripts/train_dummy.py --input_dir ${source_dir}/tfrecords/ --dataset moving_mnist --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir}/${model}_bing_20200902/ +srun python ../scripts/train_dummy.py --input_dir ${source_dir}/tfrecords/ --dataset moving_mnist --model ${model} --model_hparams_dict ${model_hparams} --output_dir ${destination_dir}/${model} -- GitLab