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

make train_movingmnist.sh

parent be598477
No related branches found
No related tags found
No related merge requests found
...@@ -4,18 +4,13 @@ ...@@ -4,18 +4,13 @@
#SBATCH --ntasks=1 #SBATCH --ntasks=1
##SBATCH --ntasks-per-node=1 ##SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=1 #SBATCH --cpus-per-task=1
<<<<<<< Updated upstream
#SBATCH --output=train_moving_mnist-out.%j #SBATCH --output=train_moving_mnist-out.%j
#SBATCH --error=train_moving_mnist-err.%j #SBATCH --error=train_moving_mnist-err.%j
======= #SBATCH --time=23:20:00
#SBATCH --output=train_era5-out.%j
#SBATCH --error=train_era5-err.%j
>>>>>>> Stashed changes
#SBATCH --time=00:20:00
#SBATCH --gres=gpu:1 #SBATCH --gres=gpu:1
#SBATCH --partition=develgpus #SBATCH --partition=gpus
#SBATCH --mail-type=ALL #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 ##jutil env activate -p cjjsc42
...@@ -48,5 +43,5 @@ model_hparams=../hparams/${dataset}/${model}/model_hparams.json ...@@ -48,5 +43,5 @@ model_hparams=../hparams/${dataset}/${model}/model_hparams.json
# rund training # 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}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment