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

Fix unstable training by limiting pytorch version to 1.1

Browse source
This commit is contained in:
Yin Li 2019-12-08 21:02:08 -05:00
parent 437126e296
commit 11c9caa1e2
7 changed files with 18 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

7
scripts/dis2dis-test.slurm
View file

@ -15,10 +15,7 @@
hostname; pwd; date
module load gcc openmpi2
module load cuda/10.1.243_418.87.00 cudnn/v7.6.2-cuda-10.1
source $HOME/anaconda3/bin/activate torch
module load gcc python3
export OMP_NUM_THREADS=$SLURM_CPUS_ON_NODE
@ -37,7 +34,7 @@ in_files="$files"
tgt_files="$files"
srun m2m.py test \
m2m.py test \
--test-in-patterns "$data_root_dir/$in_dir/$test_dirs/$in_files" \
--test-tgt-patterns "$data_root_dir/$tgt_dir/$test_dirs/$tgt_files" \
--in-channels 3 --out-channels 3 --norms cosmology.dis \

5
scripts/dis2dis.slurm
View file

@ -17,10 +17,7 @@
hostname; pwd; date
module load gcc openmpi2
module load cuda/10.1.243_418.87.00 cudnn/v7.6.2-cuda-10.1
source $HOME/anaconda3/bin/activate torch
module load gcc python3
export MASTER_ADDR=$HOSTNAME

7
scripts/vel2vel-test.slurm
View file

@ -15,10 +15,7 @@
hostname; pwd; date
module load gcc openmpi2
module load cuda/10.1.243_418.87.00 cudnn/v7.6.2-cuda-10.1
source $HOME/anaconda3/bin/activate torch
module load gcc python3
export OMP_NUM_THREADS=$SLURM_CPUS_ON_NODE
@ -37,7 +34,7 @@ in_files="$files"
tgt_files="$files"
srun m2m.py test \
m2m.py test \
--test-in-patterns "$data_root_dir/$in_dir/$test_dirs/$in_files" \
--test-tgt-patterns "$data_root_dir/$tgt_dir/$test_dirs/$tgt_files" \
--in-channels 3 --out-channels 3 --norms cosmology.vel \

5
scripts/vel2vel.slurm
View file

@ -17,10 +17,7 @@
hostname; pwd; date
module load gcc openmpi2
module load cuda/10.1.243_418.87.00 cudnn/v7.6.2-cuda-10.1
source $HOME/anaconda3/bin/activate torch
module load gcc python3
export MASTER_ADDR=$HOSTNAME