Commit Graph

151 Commits

Author SHA1 Message Date
Yin Li
d0ab596e11 Remove min-reduction; Split grad plots; Disable cgan w/o adv 2020-07-13 09:40:15 -04:00
Yin Li
29ab550032 Change instance noise
Simpler linear annealing as in the blog post;
different noise to output and target
2020-07-11 01:56:46 -04:00
Yin Li
873258d8a7 Add linear resamplers 2020-07-11 01:37:23 -04:00
Yin Li
c956dca795 Add narrow_cast and narrow_by 2020-07-11 01:24:20 -04:00
Yin Li
28ec245a7a Add pixel-shifting data augmentation 2020-07-11 00:46:19 -04:00
Yin Li
8a95d69818 Add cropping anchors controlled by start, stop, step 2020-07-11 00:39:19 -04:00
Yin Li
819e77cd86 Add slice plotting for field >2d 2020-07-10 14:58:08 -04:00
Yin Li
5c4d244a54 Add Lagrangian-to-Eulerian transformation 2020-07-10 12:52:56 -04:00
Yin Li
c3d7456d0c Change .pth to .pt following torch convention 2020-06-20 18:31:09 -04:00
Yin Li
c3877ae982 Fix README order 2020-06-14 18:19:46 -04:00
Yin Li
b1b3dbe41c Merge branch 'master' of github.com:eelregit/map2map 2020-06-14 18:00:25 -04:00
Yin Li
90a0d6e0f8 Add callback loading mechanism 2020-06-14 17:59:31 -04:00
Renan Alves de Oliveira
63c2b1a39d
add sys library 2020-06-06 15:57:01 -03:00
Yin Li
5bb2a19933 Add grouped cache for data bigger than CPU RAM 2020-05-28 23:01:56 -04:00
Yin Li
2e687da905 Add LRU cache to replace existing cache 2020-05-16 23:26:53 -04:00
Yin Li
4cc2fd51eb Refactor runtime default argument setting 2020-05-16 17:28:42 -04:00
Yin Li
9b456d6b1a Fix no data bug with --div-data 2020-05-07 21:12:05 -04:00
Yin Li
8bb6e85c6d Improve figure appearance 2020-05-07 19:18:06 -04:00
Yin Li
2c5c4fae2c Remove smooth transition halflife argument 2020-05-07 18:17:13 -04:00
Yin Li
897c3563db Add additive and multiplicative augmentation 2020-05-07 15:57:05 -04:00
Yin Li
67e5ed9eb6 Remove smooth transition to adversarial loss 2020-05-07 10:48:55 -04:00
Yin Li
f442dd59ba Remove noise channels 2020-05-06 15:33:21 -04:00
Yin Li
3e13202fb5 Shrink space between images 2020-05-04 17:13:24 -04:00
Yin Li
9b06b0f1a7 Shrink colorbar length 2020-05-04 11:32:48 -04:00
Yin Li
69e9992545 Shrink image sizes 2020-05-02 23:43:33 -04:00
Yin Li
b50c7b6350 Remove input upsampling 2020-04-30 21:12:34 -04:00
Yin Li
a88f27a3a1 Add better looking tensorboard images 2020-04-22 00:56:29 -04:00
Yin Li
996c0d3aed Add auto loading of checkpoint 2020-04-21 18:34:18 -04:00
Yin Li
c9f468c568 Add explicit barrier after init dist groups 2020-04-21 13:38:02 -04:00
Yin Li
28d7cc9e9b Add default values to command line help 2020-04-21 13:01:17 -04:00
Yin Li
356e28c31f Add TOC 2020-04-21 12:34:14 -04:00
Yin Li
5adf6cbac6 Add exit when GPU not found 2020-04-20 19:27:57 -04:00
Yin Li
facfc13132 Add descriptions on files and tensorboard 2020-04-19 14:34:23 -04:00
Yin Li
01a60cc0c7 Add checkpoint symlink to state file 2020-04-16 17:50:08 -04:00
Yin Li
d01d0cee83 Add noise annealing duration 2020-03-09 12:07:55 -04:00
Yin Li
93d973b5c8 Change state saving to every epoch 2020-03-07 18:03:19 -05:00
Yin Li
ccb323e6ee Rewrite instance noise annealing scheduler 2020-03-07 15:04:01 -05:00
Yin Li
7a0f0bd979 Add new pytorch features with fallback to old 2020-03-07 14:49:18 -05:00
Yin Li
b7b14a37eb Make main.py runnable 2020-03-04 22:23:34 -05:00
Yin Li
5fbf447e2d Fix spectral norm interference with gradient visualization
spectral norm introduces weight_orig, so that not all weight parameters
ends with ".weight"
2020-03-04 16:21:00 -05:00
Yin Li
33fd13e4ac Fix bug from 1963530 to show last instead of 2nd layer gradients 2020-03-03 20:53:34 -05:00
Yin Li
d768aadb36 Add instructions 2020-03-03 20:23:13 -05:00
Yin Li
ef993d83e9 Add instance noise 2020-03-03 20:22:31 -05:00
Yin Li
1963530984 Add gradient norms to tensorboard 2020-03-02 11:31:38 -05:00
Yin Li
9567db7332 Add weight initialization 2020-03-01 22:31:30 -05:00
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