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

Change .pth to .pt following torch convention

Browse source
This commit is contained in:
Yin Li 2020-06-20 18:31:09 -04:00
parent c3877ae982
commit c3d7456d0c
4 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
scripts/dis2dis-test.slurm
View file

@ -37,7 +37,7 @@ m2m.py test \
--test-tgt-patterns "$data_root_dir/$tgt_dir/$test_dirs/$tgt_files" \
--in-norms cosmology.dis --tgt-norms cosmology.dis --crop 256 --pad 20 \
--model VNet \
--load-state best_model.pth \
--load-state best_model.pt \
--batches 1 --loader-workers 0 \
--cache

2
scripts/vel2vel-test.slurm
View file

@ -37,7 +37,7 @@ m2m.py test \
--test-tgt-patterns "$data_root_dir/$tgt_dir/$test_dirs/$tgt_files" \
--in-norms cosmology.vel --tgt-norms cosmology.vel --crop 256 --pad 20 \
--model VNet \
--load-state best_model.pth \
--load-state best_model.pt \
--batches 1 --loader-workers 0 \
--cache