Commit Graph

153 Commits

Author SHA1 Message Date
Yin Li
9abcee4870 Replace np.take by advanced indexing for cropping 2020-07-14 14:48:54 -04:00
Yin Li
c61b6d6fab Fix bug due to strange numpy type conversion 2020-07-13 23:50:47 -04:00
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