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 |
|
Yin Li
|
de24f8d585
|
Fix testing glob patterns
|
2019-12-18 15:31:39 -05:00 |
|
Yin Li
|
796c7918d8
|
Print args in testing
|
2019-12-17 20:23:27 -05:00 |
|
Yin Li
|
843fe09a92
|
Add data caching, and new pad and crop features
|
2019-12-17 12:00:13 -05:00 |
|
Yin Li
|
d03bcb59a1
|
Fix bug from 0533150
|
2019-12-12 19:26:57 -05:00 |
|
Yin Li
|
6d021ec949
|
Simplify swish for readability
|
2019-12-12 19:24:35 -05:00 |
|
Yin Li
|
0533150194
|
Save past best models by not overwriting them
|
2019-12-12 18:05:16 -05:00 |
|
Yin Li
|
946805c6be
|
Add swish activation, kernel_size in ConvBlock, and optional trailing activation in ResBlock
|
2019-12-12 16:49:19 -05:00 |
|
Yin Li
|
341bdbff84
|
Fix issue that pytorch 1.1 does not flush tensorboard
|
2019-12-12 15:25:45 -05:00 |
|
Yin Li
|
bd3798222a
|
Add weight decay
|
2019-12-12 12:04:39 -05:00 |
|
Yin Li
|
96b683bfef
|
Fix bug in cropping width indices
|
2019-12-11 16:45:14 -05:00 |
|
Yin Li
|
0c4a551354
|
Fix input shape and norm, and target norm in testing
|
2019-12-10 17:19:57 -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
|
9e039c0407
|
Add variable skip connection in ResBlock
|
2019-12-09 20:59:39 -05:00 |
|
Yin Li
|
139caabe31
|
Remove pytorch version constraint and Add python 3 constraint in setup.py
|
2019-12-09 17:21:25 -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
|
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 |
|