Commit Graph

83 Commits

Author SHA1 Message Date
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
1cd34c2eed Remove lazy argument feeding from field dataset 2020-02-07 09:49:27 -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
Yin Li
6252629631 Change tensorboard to optional dependency 2020-01-23 09:55:01 -05:00
Yin Li
6c030f1eb1 Fix bug that target needs to be cropped with upsampled size 2020-01-23 09:29:43 -05:00
Yin Li
6938eea089 Fix bug that F.interpolate does not apply to np arrays 2020-01-23 07:22:34 -05:00
Yin Li
84a369d4ed Add super-resolution 2020-01-22 19:47:27 -05:00
Yin Li
ba42bc6a55 Add PatchGAN to __init__ 2020-01-22 19:02:44 -05:00
Yin Li
862c9e75a0 Swap generator & discriminator order in training
The reasoning is that updating generator first will free up the memory
taken by the graph of the model
2020-01-22 19:00:13 -05:00
Yin Li
f99fd6b177 Add delay epoch before updating generator with adversarial loss 2020-01-22 18:44:09 -05:00
Yin Li
cdb00ebd8d Add a simple PatchGAN 2020-01-22 14:01:18 -05:00
Yin Li
c68b9928ee Add separate input and target normalization 2020-01-21 18:51:32 -05:00
Yin Li
46a3a3a97d Fix augmentation bug 2020-01-21 18:49:45 -05:00
Yin Li
1f89e894cc Make both fat and lean V-Net available 2020-01-21 15:22:42 -05:00
Yin Li
6e06682751 Update slurm script to include adversary model 2020-01-21 12:44:46 -05:00
Yin Li
94ce018cb8 Change U-Net and V-Net inner naming convention; Slim down V-Net 2020-01-20 21:49:01 -05:00
Yin Li
01ff0aca37 Change activation default to leaky ReLU in conv block 2020-01-20 21:32:56 -05:00
Yin Li
6e48905cc0 Remove model loading hack in testing 2020-01-20 21:32:02 -05:00
Yin Li
2b7e559910 Change adversary optim scheduler to sync with model optim scheduler 2020-01-20 16:25:15 -05:00
Yin Li
e7d2435a96 Fix adversary model optim scheduler 2020-01-19 19:24:10 -05:00
Yin Li
1b1e0e82fa Fix (possibly) multiple DistributedDataParallel reduce interference 2020-01-10 14:39:16 -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
848dc87169 Change __dict__ to getattr 2019-12-23 16:11:43 -05:00
Yin Li
77710bc8a3 Cache data during the first epoch 2019-12-18 17:51:11 -05:00
Yin Li
01b0c8b514 Add data division, good with data caching 2019-12-18 17:13:40 -05:00