Commit Graph

19 Commits

Author SHA1 Message Date
Yin Li
996c0d3aed Add auto loading of checkpoint 2020-04-21 18:34:18 -04:00
Yin Li
b67079bf72 Add runtime address and port determination and share them via file
Together with slurm step node counts, make it possible to launch
multiple training in one job
2020-02-13 19:56:54 -06:00
Yin Li
1dd2b6d748 Extend training epochs in example scripts 2020-02-09 17:32:27 -05:00
Yin Li
a237245514 Add displacement to density script 2020-01-31 11:13:55 -05:00
Yin Li
ec074748d0 Fix slurm script for training on 2 nodes on Frontera 2020-01-27 11:15:46 -05:00
Yin Li
84a369d4ed Add super-resolution 2020-01-22 19:47:27 -05:00
Yin Li
c68b9928ee Add separate input and target normalization 2020-01-21 18:51:32 -05:00
Yin Li
6e06682751 Update slurm script to include adversary model 2020-01-21 12:44:46 -05:00
Yin Li
9cf97b3ac1 Fix seeding bug introduced in the completely wrong commit f64b1e4 2020-01-06 20:20:05 -05:00
Yin Li
848dc87169 Change __dict__ to getattr 2019-12-23 16:11:43 -05:00
Yin Li
01b0c8b514 Add data division, good with data caching 2019-12-18 17:13:40 -05:00
Yin Li
de24f8d585 Fix testing glob patterns 2019-12-18 15:31:39 -05:00
Yin Li
843fe09a92 Add data caching, and new pad and crop features 2019-12-17 12:00:13 -05:00
Yin Li
7b6ff73be1 Move UNet with ResBlock to VNet and Revert UNet to the previous simple version 2019-12-09 21:53:27 -05:00
Yin Li
0764a1006e Remove unnecessary arguments --in-channels and --out-channels 2019-12-09 10:19:21 -05:00
Yin Li
11c9caa1e2 Fix unstable training by limiting pytorch version to 1.1 2019-12-08 21:02:08 -05:00
Yin Li
afeefcaa9e Merge branch 'master' of github.com:eelregit/map2map
Conflicts:
	map2map/train.py
2019-12-03 17:52:01 -05:00
Yin Li
0211eed0ec Add testing 2019-12-01 21:40:35 -05:00
Yin Li
88bfd11594 Add training 2019-11-30 16:31:10 -05:00