37 lines
727 B
Bash
37 lines
727 B
Bash
#!/bin/bash
|
|
|
|
#SBATCH --job-name=R2D2
|
|
#SBATCH --output=%x-%j.out
|
|
#SBATCH --partition=cpu_partition
|
|
#SBATCH --nodes=1
|
|
#SBATCH --exclusive
|
|
##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
|
|
|
|
#module load python
|
|
#module list
|
|
|
|
#source $HOME/anaconda3/bin/activate pytorch_env
|
|
#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
|