Commit Graph

194 Commits

Author SHA1 Message Date
Yin Li
9c8b331bf5 Add loglxe loss as log of product of Lagrangian and Eulerian losses 2020-08-09 12:59:24 -07:00
Yin Li
13edf3b96d Add optional trailing activation to residual block 2020-08-09 12:18:17 -07:00
Yin Li
a1e5399311 Fix args helps 2020-08-04 21:15:22 -07:00
Yin Li
6b65b7888d Change color normalization in plt_slices 2020-08-04 21:13:13 -07:00
Yin Li
d6b7a528df Remove pyramid model 2020-08-04 21:13:13 -07:00
Yin Li
728e0597f0 Add documentation 2020-08-04 21:13:13 -07:00
Yin Li
265587922d Add srsgan model and scale_factor to model arguments 2020-08-04 21:13:10 -07:00
Yin Li
6dfde5ee7f Fix --optimizer-args None default bug 2020-07-31 17:31:55 -04:00
Yin Li
5ac9016987 Replace specific slurm scripts with general ones 2020-07-29 12:12:46 -07:00
Yin Li
b54fc4ba3a Add a line about socket hopefully will avoid port conflicts 2020-07-28 07:38:07 -07:00
Yin Li
0d41bdae26 Add plt.close to avoid matplotlib warning 2020-07-28 07:10:46 -07:00
Yin Li
a886e53c54 Always use training loss for lr scheduler 2020-07-27 17:47:58 -07:00
Yin Li
2336d83f2d Add optimizer and scheduler kwargs via json.loads 2020-07-27 12:40:19 -07:00
Yin Li
8ce13e67f6 Fix training hang due to constrained layout of matplotlib 2020-07-27 12:39:41 -07:00
Yin Li
bf9b0ba426 Merge remote-tracking branch 'refs/remotes/origin/lag2eul' into lag2eul 2020-07-22 18:48:27 -07:00
Yin Li
ca39b2028f Replace cuda.device with env var to assign GPU to processes 2020-07-22 18:48:12 -07:00
Yin Li
f98b0fbbd0 Merge remote-tracking branch 'refs/remotes/origin/lag2eul' into lag2eul 2020-07-22 02:38:45 -04:00
Yin Li
98cdd4795c Add DistFieldSampler to benefit from page cache 2020-07-21 23:37:59 -07:00
Yin Li
632c73db16 Merge branch 'lag2eul' of github.com:eelregit/map2map into lag2eul 2020-07-22 00:20:28 -04:00
Yin Li
9f54e02c3a Add DistFieldSampler to benefit from page cache 2020-07-21 21:13:52 -07:00
Yin Li
ec46f41ba5 Add lag x eul loss tracking to tensorboard 2020-07-21 21:10:34 -07:00
Yin Li
509856cb4f Revert "Deprecate GroupedRandomSampler for use of memmap in place of cache"
This reverts commit bbf77c9f91.
2020-07-18 18:07:38 -04:00
Yin Li
56f8fa932b Revert "Deprecate GroupedRandomSampler for use of memmap in place of cache"
This reverts commit bbf77c9f91.
2020-07-16 15:22:52 -07:00
Yin Li
5bc099251f Merge remote-tracking branch 'refs/remotes/origin/lag2eul' into lag2eul 2020-07-15 19:04:34 -07:00
Yin Li
2d5234812b Add lagrangian and eulerian alternate training 2020-07-15 22:03:30 -04:00
Yin Li
337d65de68 Remove adversary 2020-07-15 22:03:30 -04:00
Yin Li
607bcf3f4c Add Lag2Eul to training 2020-07-15 22:03:30 -04:00
Yin Li
40220e9248 Add Lagrangian-to-Eulerian transformation 2020-07-15 22:03:30 -04:00
Yin Li
ff4c2e12fd Change default number of loader workers 2020-07-15 22:03:14 -04:00
Yin Li
7d3a598080 Remove shape dependence from narrow_by
Good for torchscript
2020-07-15 21:36:37 -04:00
Yin Li
4160ed25f7 Add lagrangian and eulerian alternate training 2020-07-15 15:57:10 -04:00
Yin Li
e8744d6c7b Remove adversary 2020-07-14 21:07:05 -04:00
Yin Li
6fa85f8285 Merge branch 'master' into lag2eul 2020-07-14 18:47:54 -04:00
Yin Li
bbf77c9f91 Deprecate GroupedRandomSampler for use of memmap in place of cache 2020-07-14 18:47:26 -04:00
Yin Li
3437b20ed8 Merge branch 'master' into lag2eul 2020-07-14 18:28:10 -04:00
Yin Li
5fd8cbfcda Change default number of loader workers 2020-07-14 18:16:25 -04:00
Yin Li
818ed6923d Add memmap to numpy data loading 2020-07-14 17:42:10 -04:00
Yin Li
eba76bf90d Merge branch 'master' into memmap 2020-07-14 14:52:03 -04:00
Yin Li
9abcee4870 Replace np.take by advanced indexing for cropping 2020-07-14 14:48:54 -04:00
Yin Li
c2edeb344e Replace np.take by advanced indexing for cropping 2020-07-14 14:33:53 -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
bab3f08aa5 Add Lag2Eul to training 2020-07-10 12:50:07 -04:00