diff --git a/scripts/example-test.slurm b/scripts/example-test.slurm index 1036450..4e7398b 100644 --- a/scripts/example-test.slurm +++ b/scripts/example-test.slurm @@ -14,8 +14,12 @@ hostname; pwd; date # set computing environment, e.g. with module or anaconda + #module load python +#module list + #source $HOME/anaconda3/bin/activate pytorch_env +#conda info export OMP_NUM_THREADS=$SLURM_CPUS_ON_NODE # use MKL-DNN diff --git a/scripts/example-train.slurm b/scripts/example-train.slurm index 3cd4049..d6cc894 100644 --- a/scripts/example-train.slurm +++ b/scripts/example-train.slurm @@ -26,8 +26,13 @@ hostname; pwd; date # set computing environment, e.g. with module or anaconda + #module load python +#module list + #source $HOME/anaconda3/bin/activate pytorch_env +#conda info + srun m2m.py train \ --train-in-patterns "train/R0-*.npy,train/R1-*.npy" \