Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add data caching, and new pad and crop features

Browse source
This commit is contained in:
Yin Li 2019-12-17 12:00:13 -05:00
parent d03bcb59a1
commit 843fe09a92
6 changed files with 113 additions and 68 deletions
Download patch file Download diff file Expand all files Collapse all files

6
scripts/dis2dis-test.slurm
View file

@ -29,7 +29,7 @@ tgt_dir="nonlin"
test_dirs="0" # FIXME
files="dis/128x???.npy"
files="dis/512x???.npy"
in_files="$files"
tgt_files="$files"
@ -37,8 +37,8 @@ tgt_files="$files"
m2m.py test \
--test-in-patterns "$data_root_dir/$in_dir/$test_dirs/$in_files" \
--test-tgt-patterns "$data_root_dir/$tgt_dir/$test_dirs/$tgt_files" \
--norms cosmology.dis --model VNet \
--batches 1 --loader-workers 0 --pad-or-crop 40 \
--norms cosmology.dis --model VNet --cache --crop 128 --pad 50 \
--batches 1 --loader-workers 0 \
--load-state best_model.pth

6
scripts/vel2vel-test.slurm
View file

@ -29,7 +29,7 @@ tgt_dir="nonlin"
test_dirs="0" # FIXME
files="vel/128x???.npy"
files="vel/512x???.npy"
in_files="$files"
tgt_files="$files"
@ -37,8 +37,8 @@ tgt_files="$files"
m2m.py test \
--test-in-patterns "$data_root_dir/$in_dir/$test_dirs/$in_files" \
--test-tgt-patterns "$data_root_dir/$tgt_dir/$test_dirs/$tgt_files" \
--norms cosmology.vel --model VNet \
--batches 1 --loader-workers 0 --pad-or-crop 40 \
--norms cosmology.vel --model VNet --cache --crop 128 --pad 50 \
--batches 1 --loader-workers 0 \
--load-state best_model.pth