Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix super-resolution script for Frontera

Browse Source
This commit is contained in:
Yin Li 2020-01-23 11:53:53 -06:00
parent 1b3f32b791
commit 0d1827669a
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/srsgan.slurm
View File

@ -4,18 +4,19 @@
#SBATCH --output=%x-%j.out #SBATCH --output=%x-%j.out
#SBATCH --partition=rtx #SBATCH --partition=rtx
#SBATCH --gres=gpu:4 ##SBATCH --gres=gpu:4
#SBATCH --exclusive #SBATCH --exclusive
#SBATCH --nodes=1 #SBATCH --nodes=1
#SBATCH --time=7-00:00:00 #SBATCH --ntasks-per-node=1
#SBATCH --time=2-00:00:00
hostname; pwd; date hostname; pwd; date
#module load gcc python3 #module load gcc python3
source $HOME/anaconda/bin/activate torch source $HOME/anaconda3/bin/activate
export MASTER_ADDR=$HOSTNAME export MASTER_ADDR=$HOSTNAME
@ -39,13 +40,13 @@ tgt_files_2="vel.npy"
srun m2m.py train \ 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-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" \ --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 \ --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 \ --model VNet --adv-model PatchGAN --cgan \
--lr 0.0001 --adv-lr 0.0004 --batches 1 --loader-workers 0 \ --lr 0.0001 --adv-lr 0.0004 --batches 1 --loader-workers 0 \
--epochs 128 --seed $RANDOM \ --epochs 128 --seed $RANDOM \
--cache --div-data --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 \ # --load-state checkpoint.pth \