Commit Graph

206 Commits

Author SHA1 Message Date
Yin Li
75b1c19dcd Combine input and output figures 2020-03-01 15:40:53 -05:00
Yin Li
a46746287a Fix bug when env SLURM_STEP_NUM_NODES is not defined 2020-02-14 15:51:36 -05:00
Yin Li
5cb4a1bbae Remove label smoothing for generator 2020-02-14 09:22:57 -05: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
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
e8039dcccc Move map loss computation forward in training 2020-02-13 14:40:51 -05:00
Yin Li
e383ec3977 Fix warnings on incompatible keys 2020-02-13 14:39:37 -05:00
Yin Li
53ed5a91f4 Change to nearest interpolation from linear in model super-resolution
To be consistent with the data super-resolution in data/fields.py
2020-02-13 12:01:04 -05:00
Yin Li
db3414e11c Disable D grad when updating G; Revert D and G order
The reason for the latter is to follow most code but I haven't
found the reason
2020-02-13 11:33:40 -05:00
Yin Li
aeeeb966d8 Add stride to ConvBlock 2020-02-11 17:20:42 -05:00
Yin Li
cc0efd28ec Fix bug: pprint -> pformat 2020-02-11 12:55:08 -05:00
Yin Li
2e2adc761d Add spectral normalization 2020-02-11 11:58:05 -05:00
Yin Li
b41e85eda5 Fix stupid merging bug 2020-02-09 23:07:40 -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
a2e3a9dc64 Change PatchGAN architecture 2020-02-09 17:33:23 -05:00
Yin Li
1dd2b6d748 Extend training epochs in example scripts 2020-02-09 17:32:27 -05:00
Yin Li
09493ad4ec Add non-strict model state loading 2020-02-09 17:32:09 -05:00
Yin Li
f831afbccf Add PatchGAN42 similar to the PatchGAN in pix2pix 2020-02-07 15:26:16 -05:00
Yin Li
45c1d57e72 Add pretty-print argument to job stdout 2020-02-07 11:40:08 -05:00
Yin Li
3fb9708575 Remove saving / loading optimizer & scheduler states
Conflicts:
	map2map/train.py
2020-02-07 10:15:14 -05:00
Yin Li
23b903f81b Remove saving / loading optimizer & scheduler states 2020-02-07 10:02:32 -05:00
Yin Li
1cd34c2eed Remove lazy argument feeding from field dataset 2020-02-07 09:49:27 -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
291dfb24b3 Add diff between out and tgt to figure 2020-02-05 21:19:03 -05:00
Yin Li
e20f3194a5 Change linear interpolation to nearest in super-resolution 2020-02-05 21:09:39 -05:00
Yin Li
679f9f2545 Add pyramid network 2020-02-05 20:20:17 -05:00
Yin Li
a5c48e71b0 Change noise channel removal from input 2020-02-05 14:40:45 -05:00
Yin Li
ef0235f97b Fix upsampling crop bug 2020-02-05 14:10:58 -05:00
Yin Li
b609797e27 Add per field colormap and Fix scope bug 2020-02-05 11:58:05 -05:00
Yin Li
a22fb64d12 Wrap Dice loss with nn.Module 2020-02-03 22:19:38 -05:00
Yin Li
db69e9f953 Add figures with tensorboard 2020-02-03 22:18:08 -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
a237245514 Add displacement to density script 2020-01-31 11:13:55 -05:00
Yin Li
ea61bb9b65 Add exp and log functions from torch 2020-01-31 11:11:31 -05:00
Yin Li
698b2a8df7 Add Dice loss 2020-01-29 15:58:07 -05:00
Yin Li
0721301113 Add noise channels to the input 2020-01-29 15:57:10 -05:00
Yin Li
31ea70fca9 Merge branch 'master' of github.com:eelregit/map2map
Conflicts:
	scripts/srsgan.slurm
2020-01-27 11:17:24 -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
0d1827669a Fix super-resolution script for Frontera 2020-01-23 11:53:53 -06:00
Yin Li
1b3f32b791 Merge branch 'master' of github.com:eelregit/map2map 2020-01-23 11:35:57 -06:00
Yin Li
71bf37d0aa Merge branch 'master' of github.com:eelregit/map2map 2020-01-23 11:35:26 -06:00
Yin Li
de71df51f5 Fix touch-up... 2020-01-23 12:34:32 -05:00
yueyingn
d5b76b3725
fix crop function with upsample 2020-01-23 11:34:20 -05:00
Yin Li
1008312a20 Merge branch 'master' of github.com:eelregit/map2map 2020-01-23 09:00:33 -06:00
Yin Li
55ee131910 Merge 2020-01-23 09:00:31 -06:00
Yin Li
581ad5d60b Change tensorboard to optional dependency 2020-01-23 09:58:54 -05:00