Commit Graph

227 Commits

Author SHA1 Message Date
Yin Li
4be71a32d1 Add module and conda env recording 2021-03-18 15:27:48 -04:00
Yin Li
8544ff07e8 Fixes and cleaning up 2021-03-18 14:24:44 -04:00
Yin Li
55b1a72ef4 Add kernel_size and stride to ResBlock 2021-03-18 14:21:20 -04:00
Yin Li
fd1cdb0ce7 Add unequal/asymmetric cropping, padding, and aug_shift 2021-03-18 14:21:20 -04:00
Yin Li
183a223ee6 Fix error type 2021-03-18 14:21:20 -04:00
Yin Li
c102fc588e Add cmap and norm for all non positive images in plt_slices 2020-09-23 18:26:40 -04:00
Yin Li
e40ea52190 Change crop, flip, perm to in-place 2020-09-23 18:12:42 -04:00
Yin Li
156011be5f Fix indentation bug pointed out by Patrick Lachance 2020-09-21 19:56:50 -04:00
Yin Li
7971137ad4 Broaden linear region of SymLogNorm in plt_slices 2020-09-21 19:56:29 -04:00
Yin Li
3f35a5cca6 Add identity normalization 2020-09-21 19:56:29 -04:00
Yin Li
b7a9adf0c7 Add pytorch version reporting 2020-09-21 19:56:29 -04:00
Yin Li
a579c9655b Replace assert with raise 2020-09-21 19:56:29 -04:00
Yin Li
55f61b9a70 Add optional optimizer/scheduler loading 2020-09-12 19:27:31 -04:00
Yin Li
b215c0fa7b Fix import_attr bug pointed out by David Schaurecker 2020-09-12 18:48:44 -04:00
Yin Li
4799f8177c Fix to rename --batches to --batch-size, former kept for BC 2020-09-12 18:30:14 -04:00
Yin Li
7be3153206 Revert removal of saving/loading optimizer & scheduler states
Why did I do that?
2020-09-12 18:30:07 -04:00
Yin Li
85efb9e3a3 Add autograd detect anomaly flag, and test cudnn.deterministic
cudnn.deterministic is only 10% slower
2020-09-12 18:26:44 -04:00
Yin Li
d8c6be797d Move figures from .data to .utils 2020-09-12 18:25:58 -04:00
Yin Li
154376d95a Add tgt_pad, rename pad to in_pad
tgt_pad can be useful for scale_factor > 1
2020-09-12 18:24:36 -04:00
Yin Li
39ad59436e Replace narrow_like by narrow_by in UNet/VNet
This makes it traceable / scriptable.
Note that the narrow_like in ResBlock used by VNet is not changed yet
2020-09-11 00:26:14 -04:00
Yin Li
c4ab7e065b Add plt_slices(size=None) 2020-09-09 14:56:07 -04:00
Yin Li
6c62cb09db Fix error pickling normalization function in multiprocessing dataloading
By avoiding importing functions as class attributes but importing before
calling, and using cache in the importing function

Thanks to help by Yu Feng
2020-09-09 14:56:07 -04:00
Yin Li
0e0ad8f071 Fix import bug that prevents pickling of norm functions
https://docs.python.org/3/library/importlib.html#importing-a-source-file-directly
https://stackoverflow.com/questions/43819653/pickling-objects-imported-with-importlib-util/53696167#53696167
2020-09-09 14:56:06 -04:00
Yin Li
f3216f4cbd Add multiple pkgs to import_attr 2020-09-09 14:55:36 -04:00
Yin Li
79b28561d5 Fix two power spectrum bugs about freq shift and flatten 2020-08-23 15:50:24 -04:00
Yin Li
6c67eaa788 Fix cropping bug when scale_factor>1 2020-08-23 15:50:24 -04:00
Yin Li
670364e54c Change Lag2Eul to lag2eul as a function 2020-08-22 23:25:08 -04:00
Yin Li
3eb1b0bccc Add power spectrum tracking 2020-08-22 23:24:25 -04:00
Yin Li
afaf4675fe Add power spectrum module 2020-08-22 23:18:48 -04:00
Yin Li
5d22594ede Add shape report after narrow_cast in training 2020-08-22 12:28:08 -07:00
Yin Li
01cc1b6964 Change VNet based on experiment on displacement 2020-08-17 19:28:10 -07:00
Yin Li
ebd962e333 Fix module before __init__ bug 2020-08-12 09:10:58 -07:00
Yin Li
61ca400942 Add center slicing in plt_slices
Also fix bug of hard coded `size`
2020-08-11 13:05:38 -07:00
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