map2map/scripts/example-test.slurm

37 lines
727 B
Plaintext
Raw Normal View History

#!/bin/bash
#SBATCH --job-name=R2D2
#SBATCH --output=%x-%j.out
#SBATCH --partition=cpu_partition
2021-05-31 04:51:59 +02:00
#SBATCH --nodes=1
#SBATCH --exclusive
2021-05-31 04:51:59 +02:00
##SBATCH --partition=gpu_partition
##SBATCH --gres=gpu:1
##SBATCH --ntasks=1
##SBATCH --cpus-per-task=8
#SBATCH --time=0-01:00:00
hostname; pwd; date
# set computing environment, e.g. with module or anaconda
2021-03-18 20:17:30 +01:00
#module load python
2021-03-18 20:17:30 +01:00
#module list
#source $HOME/anaconda3/bin/activate pytorch_env
2021-03-18 20:17:30 +01:00
#conda info
m2m.py test \
--test-in-patterns "test/R0-*.npy,test/R1-*.npy" \
--test-tgt-patterns "test/D0-*.npy,test/D1-*.npy" \
--in-norms RnD.R0,RnD.R1 --tgt-norms RnD.D0,RnD.D1 \
--model model.Net --callback-at . \
--batch-size 1 \
--load-state checkpoint.pt
date