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

Add module and conda env recording

Browse Source
This commit is contained in:
Yin Li 2021-03-18 15:17:30 -04:00
parent 8544ff07e8
commit 4be71a32d1
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/example-test.slurm
View File

@ -14,8 +14,12 @@ hostname; pwd; date
# set computing environment, e.g. with module or anaconda # set computing environment, e.g. with module or anaconda
#module load python #module load python
#module list
#source $HOME/anaconda3/bin/activate pytorch_env #source $HOME/anaconda3/bin/activate pytorch_env
#conda info
export OMP_NUM_THREADS=$SLURM_CPUS_ON_NODE # use MKL-DNN export OMP_NUM_THREADS=$SLURM_CPUS_ON_NODE # use MKL-DNN

5
scripts/example-train.slurm
View File

@ -26,8 +26,13 @@ hostname; pwd; date
# set computing environment, e.g. with module or anaconda # set computing environment, e.g. with module or anaconda
#module load python #module load python
#module list
#source $HOME/anaconda3/bin/activate pytorch_env #source $HOME/anaconda3/bin/activate pytorch_env
#conda info
srun m2m.py train \ srun m2m.py train \
--train-in-patterns "train/R0-*.npy,train/R1-*.npy" \ --train-in-patterns "train/R0-*.npy,train/R1-*.npy" \