From 0d1827669a0c7c78d2e6f0fae76a6381c8b9aff6 Mon Sep 17 00:00:00 2001 From: Yin Li Date: Thu, 23 Jan 2020 11:53:53 -0600 Subject: [PATCH] Fix super-resolution script for Frontera --- scripts/srsgan.slurm | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/scripts/srsgan.slurm b/scripts/srsgan.slurm index 0d9e265..92390cf 100644 --- a/scripts/srsgan.slurm +++ b/scripts/srsgan.slurm @@ -4,18 +4,19 @@ #SBATCH --output=%x-%j.out #SBATCH --partition=rtx -#SBATCH --gres=gpu:4 +##SBATCH --gres=gpu:4 #SBATCH --exclusive #SBATCH --nodes=1 -#SBATCH --time=7-00:00:00 +#SBATCH --ntasks-per-node=1 +#SBATCH --time=2-00:00:00 hostname; pwd; date #module load gcc python3 -source $HOME/anaconda/bin/activate torch +source $HOME/anaconda3/bin/activate export MASTER_ADDR=$HOSTNAME @@ -39,13 +40,13 @@ tgt_files_2="vel.npy" srun m2m.py train \ --train-in-patterns "$data_root_dir/$in_dir/$train_dirs/$in_files_1,$data_root_dir/$in_dir/$train_dirs/$in_files_2" \ --train-tgt-patterns "$data_root_dir/$tgt_dir/$train_dirs/$tgt_files_1,$data_root_dir/$tgt_dir/$train_dirs/$tgt_files_2" \ - --val-in-patterns "$data_root_dir/$in_dir/$val_dirs/$in_files_1,$data_root_dir/$in_dir/$val_dirs/$in_files_2" \ - --val-tgt-patterns "$data_root_dir/$tgt_dir/$val_dirs/$tgt_files_1,$data_root_dir/$tgt_dir/$val_dirs/$tgt_files_2" \ --in-norms cosmology.dis,cosmology.vel --tgt-norms cosmology.dis,cosmology.vel --augment --crop 88 --pad 20 --scale-factor 2 \ --model VNet --adv-model PatchGAN --cgan \ --lr 0.0001 --adv-lr 0.0004 --batches 1 --loader-workers 0 \ --epochs 128 --seed $RANDOM \ --cache --div-data +# --val-in-patterns "$data_root_dir/$in_dir/$val_dirs/$in_files_1,$data_root_dir/$in_dir/$val_dirs/$in_files_2" \ +# --val-tgt-patterns "$data_root_dir/$tgt_dir/$val_dirs/$tgt_files_1,$data_root_dir/$tgt_dir/$val_dirs/$tgt_files_2" \ # --load-state checkpoint.pth \