Commit Graph

244 Commits

Author SHA1 Message Date
Yin Li
9d2cd5383b Add Lagrangian-to-Eulerian transformation 2020-07-10 12:48:39 -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
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