Yin Li
11c9caa1e2
Fix unstable training by limiting pytorch version to 1.1
2019-12-08 21:02:08 -05:00
Yin Li
437126e296
Fix DistributedDataParallel model save and load during training, leave testing for later
2019-12-08 21:00:51 -05:00
Yin Li
f2e9af6d5f
Revert scheduler to ReduceLROnPlateau
2019-12-08 20:58:46 -05:00
Yin Li
34b8c62c76
Merge branch 'master' of github.com:eelregit/map2map
...
Conflicts:
map2map/train.py
2019-12-03 18:07:11 -05:00
Yin Li
b253bb687b
Change ReduceLROnPlateau to CyclicLR
2019-12-03 18:06:33 -05:00
Yin Li
054ccb194e
Merge branch 'master' of github.com:eelregit/map2map
2019-12-03 18:03:41 -05:00
Yin Li
044e4c9376
Change ReduceLROnPlateau to CyclicLR
2019-12-03 18:02:56 -05:00
Yin Li
1f6b34dbf6
Merge branch 'master' of github.com:eelregit/map2map
2019-12-03 18:02:03 -05:00
Yin Li
36cf2ac93b
Change ReduceLROnPlateau to CyclicLR
2019-12-03 17:53:20 -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
7540154eba
Change ReduceLROnPlateau to CyclicLR
2019-12-03 17:49:56 -05:00
Yin Li
bf3fe86afe
Change ReduceLROnPlateau to CyclicLR
2019-12-03 17:40:08 -05:00
Yin Li
0211eed0ec
Add testing
2019-12-01 21:40:35 -05:00
Yin Li
bcf95275f3
Fix global_step in tensorboard summary to start from 1
2019-11-30 22:15:10 -05:00
Yin Li
9d4b5daae3
Add enhanced model
...
Add ResBlock using ConvBlock;
Use ResBlock in UNet;
Use PReLU instead of LeakyReLU;
Add more channels in lower levels in UNet;
Add more blocks in each level in UNet;
2019-11-30 21:49:10 -05:00
Yin Li
88bfd11594
Add training
2019-11-30 16:31:10 -05:00
Yin Li
6015dd6b31
Initial commit
2019-11-30 15:27:25 -05:00
