diff --git a/video_prediction_savp/HPC_scripts/train_movingmnist.sh b/video_prediction_savp/HPC_scripts/train_movingmnist.sh index 6555bb12dc0d89ab085a9567c9a9a89acc8a1d6f..b0ec5e83cf1bcdacfd5abde36be294dc560a6da5 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}