2019-12-01 18:53:38 -05:00
|
|
|
#!/bin/bash
|
|
|
|
|
|
|
|
|
|
#SBATCH --job-name=vel2vel-test
|
|
|
|
|
#SBATCH --output=%x-%j.out
|
|
|
|
|
|
|
|
|
|
#SBATCH --partition=ccm
|
|
|
|
|
|
|
|
|
|
#SBATCH --exclusive
|
|
|
|
|
#SBATCH --nodes=1
|
|
|
|
|
#SBATCH --time=1-00:00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
hostname; pwd; date
|
|
|
|
|
|
|
|
|
|
|
2020-01-31 11:13:55 -05:00
|
|
|
module load gcc python3
|
|
|
|
|
#source $HOME/anaconda3/bin/activate torch
|
2019-12-01 18:53:38 -05: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 14:37:48 -05:00
|
|
|
test_dirs="*99"
|
2019-12-01 18:53:38 -05:00
|
|
|
|
2020-01-06 20:20:05 -05:00
|
|
|
files="vel.npy"
|
2019-12-01 18:53:38 -05:00
|
|
|
in_files="$files"
|
|
|
|
|
tgt_files="$files"
|
|
|
|
|
|
|
|
|
|
|
2019-12-08 21:02:08 -05:00
|
|
|
m2m.py test \
|
2019-12-01 18:53:38 -05: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" \
|
2020-01-21 18:51:32 -05:00
|
|
|
--in-norms cosmology.vel --tgt-norms cosmology.vel --crop 256 --pad 20 \
|
2019-12-18 14:37:48 -05:00
|
|
|
--model VNet \
|
2020-06-20 18:31:09 -04:00
|
|
|
--load-state best_model.pt \
|
2020-07-14 17:05:30 -04:00
|
|
|
--batches 1 --loader-workers 0
|
2019-12-01 18:53:38 -05:00
|
|
|
|
|
|
|
|
|
|
|
|
|
date
|
