map2map/scripts/dis2dis-test.slurm

46 lines
797 B
Plaintext
Raw Normal View History

2019-12-02 00:53:38 +01:00
#!/bin/bash
#SBATCH --job-name=dis2dis-test
#SBATCH --output=%x-%j.out
#SBATCH --partition=ccm
#SBATCH --exclusive
#SBATCH --nodes=1
#SBATCH --time=1-00:00:00
hostname; pwd; date
#module load gcc python3
source $HOME/anaconda/bin/activate torch
2019-12-02 00:53:38 +01:00
export OMP_NUM_THREADS=$SLURM_CPUS_ON_NODE
data_root_dir="/mnt/ceph/users/yinli/Quijote"
in_dir="linear"
tgt_dir="nonlin"
2019-12-18 20:37:48 +01:00
test_dirs="*99"
2019-12-02 00:53:38 +01:00
files="dis.npy"
2019-12-02 00:53:38 +01:00
in_files="$files"
tgt_files="$files"
m2m.py test \
2019-12-02 00:53:38 +01:00
--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-norms cosmology.dis --tgt-norms cosmology.dis --crop 256 --pad 20 \
2019-12-18 20:37:48 +01:00
--model VNet \
--load-state best_model.pth \
--batches 1 --loader-workers 0 \
2019-12-18 20:37:48 +01:00
--cache
2019-12-02 00:53:38 +01:00
date