Commit Graph

28 Commits

Author SHA1 Message Date
Yin Li
0d4ae3424e Add cuda backend to inference 2021-05-30 23:04:50 -04:00
Yin Li
e20a3e3f62 Cleanup, Enable plt_power by default 2021-05-30 22:58:33 -04:00
Yin Li
8a3cd1843d Move the executable script to the right place
so that calling it runs the local map2map
2021-05-18 13:29:13 -04:00
Yin Li
4be71a32d1 Add module and conda env recording 2021-03-18 15:27:48 -04:00
Yin Li
4799f8177c Fix to rename --batches to --batch-size, former kept for BC 2020-09-12 18:30:14 -04:00
Yin Li
5ac9016987 Replace specific slurm scripts with general ones 2020-07-29 12:12:46 -07:00
Yin Li
e8744d6c7b Remove adversary 2020-07-14 21:07:05 -04:00
Yin Li
818ed6923d Add memmap to numpy data loading 2020-07-14 17:42:10 -04:00
Yin Li
c3d7456d0c Change .pth to .pt following torch convention 2020-06-20 18:31:09 -04:00
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