Commit graph

26 commits

Author SHA1 Message Date
Yin Li
5cb4a1bbae Remove label smoothing for generator 2020-02-14 09:22:57 -05:00
Yin Li
1818e11265 Add simple label smoothing by --adv-real-label 2020-02-13 16:44:56 -05:00
Yin Li
d2840f01b0 Add --adv-start as epoch to start adversarial training
This is similar to the deprecated --adv-delay, but specify
absolute epoch (--adv-delay counts from start_epoch in resumed training)
2020-02-13 16:00:50 -05:00
Yin Li
2e2adc761d Add spectral normalization 2020-02-11 11:58:05 -05:00
Yin Li
16b82fcc56 Merge branch 'anneal_loss'
Conflicts:
	map2map/models/patchgan.py
	map2map/train.py
2020-02-09 21:25:39 -05:00
Yin Li
2aed00d97e Add argument to disable ReduceLROnPlateau by default 2020-02-09 20:15:23 -05:00
Yin Li
09493ad4ec Add non-strict model state loading 2020-02-09 17:32:09 -05:00
Yin Li
890a459363 Fix bug 2020-02-06 19:34:43 -06:00
Yin Li
cd63324724 Change delayed adversary to a smooth annealing scheme 2020-02-06 19:04:30 -05:00
Yin Li
7f6578c63e Add delay to scheduler by adv_delay 2020-02-03 14:44:14 -05:00
Yin Li
0a2fc9a9e9 Add wrappers of adversary model and adversarial loss
commit c0dafec94bb7d131938650027f84e5308bf16ffd
Author: Yin Li <eelregit@gmail.com>
Date:   Mon Feb 3 11:18:08 2020 -0600

    Fix bug

commit b470b873649515f4b8a1cac7b4b33181eac51199
Author: Yin Li <eelregit@gmail.com>
Date:   Mon Feb 3 09:39:08 2020 -0600

    Fix bug

commit 9f8f64b3510c72bfcf2a1236ba5285edf280701c
Author: Yin Li <eelregit@gmail.com>
Date:   Mon Feb 3 10:20:37 2020 -0500

    Add wrappers of adversary model and adversarial loss
2020-02-03 12:05:47 -06:00
Yin Li
0721301113 Add noise channels to the input 2020-01-29 15:57:10 -05:00
Yin Li
84a369d4ed Add super-resolution 2020-01-22 19:47:27 -05:00
Yin Li
f99fd6b177 Add delay epoch before updating generator with adversarial loss 2020-01-22 18:44:09 -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
15384dc9bd Add optional adversary model and make validation optional 2020-01-09 20:24: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
01b0c8b514 Add data division, good with data caching 2019-12-18 17:13:40 -05:00
Yin Li
843fe09a92 Add data caching, and new pad and crop features 2019-12-17 12:00:13 -05:00
Yin Li
bd3798222a Add weight decay 2019-12-12 12:04:39 -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
f64b1e42e9 Add synchronized random seed to training 2019-12-08 21:27:44 -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