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 |
|