Yin Li
|
9d3253ac48
|
Add assemble to FieldDataset to undo cropping
|
2021-03-24 12:35:52 -04:00 |
|
Yin Li
|
89e8651c26
|
Fix and improve SRSGAN
|
2021-03-18 15:27:48 -04:00 |
|
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 |
|