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

Add auto loading of checkpoint

Browse source
This commit is contained in:
Yin Li 2020-04-21 18:29:57 -04:00
parent c9f468c568
commit 996c0d3aed
6 changed files with 37 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

1
scripts/dis2den.slurm
View file

@ -40,7 +40,6 @@ srun m2m.py train \
--lr 0.0001 --batches 1 --loader-workers 0 \
--epochs 1024 --seed $RANDOM \
--cache --div-data
# --load-state checkpoint.pth \
date

1
scripts/dis2dis.slurm
View file

@ -41,7 +41,6 @@ srun m2m.py train \
--lr 0.0001 --adv-lr 0.0004 --batches 1 --loader-workers 0 \
--epochs 1024 --seed $RANDOM \
--cache --div-data
# --load-state checkpoint.pth \
date

1
scripts/srsgan.slurm
View file

@ -43,7 +43,6 @@ srun m2m.py train \
--cache --div-data
# --val-in-patterns "$data_root_dir/$in_dir/$val_dirs/$in_files_1,$data_root_dir/$in_dir/$val_dirs/$in_files_2" \
# --val-tgt-patterns "$data_root_dir/$tgt_dir/$val_dirs/$tgt_files_1,$data_root_dir/$tgt_dir/$val_dirs/$tgt_files_2" \
# --load-state checkpoint.pth \
date

1
scripts/vel2vel.slurm
View file

@ -41,7 +41,6 @@ srun m2m.py train \
--lr 0.0001 --adv-lr 0.0004 --batches 1 --loader-workers 0 \
--epochs 1024 --seed $RANDOM \
--cache --div-data
# --load-state checkpoint.pth \
date