Yin Li
|
8a3cd1843d
|
Move the executable script to the right place
so that calling it runs the local map2map
|
2021-05-18 13:29:13 -04:00 |
|
Yin Li
|
413221af7b
|
Add matplotlib and tensorboard requirements
|
2021-05-06 12:39:27 -07:00 |
|
Yin Li
|
2717269a8d
|
Remove many item() to avoid unnecessary cpu-gpu sync
|
2021-04-23 11:40:59 -04:00 |
|
Yin Li
|
5e4633b125
|
Fix again rfft call broken by pytorch breaking BC
|
2021-04-22 13:34:04 -04:00 |
|
Yin Li
|
795cefe38e
|
Fix crop and pad are for the shapes after perm()
not before it!
This only matters for non-cubic crop and pad
|
2021-04-12 15:04:10 -07:00 |
|
Yin Li
|
08c805e47a
|
Remove size argument from crop
|
2021-04-12 17:21:33 -04:00 |
|
Yin Li
|
002e249925
|
Fix rfft call broken by pytorch breaking BC
|
2021-04-12 17:21:33 -04:00 |
|
Yin Li
|
4898724ac1
|
Remove tab
|
2021-03-26 12:50:49 -04:00 |
|
Yin Li
|
abb16fe26a
|
Improve FieldDataset.assemble
|
2021-03-26 12:47:38 -04:00 |
|
Yin Li
|
0410435a8a
|
Change default number of dataloader workers
as suggested by pytorch
|
2021-03-26 12:29:06 -04:00 |
|
Yin Li
|
3eaca964ed
|
Add styles to test.py
|
2021-03-24 15:13:21 -04:00 |
|
Yin Li
|
01c2e45430
|
Fixes and cleaning following 183a223 55b1a72 8544ff0
|
2021-03-24 12:36:07 -04:00 |
|
Yin Li
|
a61990ee45
|
Rename param to style
|
2021-03-24 12:36:07 -04:00 |
|
Yin Li
|
17d8e95870
|
Add styled VNet and Fix bugs
Co-authored-by: Drew Jamieson <drew.s.jamieson@gmail.com>
|
2021-03-24 12:36:07 -04:00 |
|
Yin Li
|
f5bd657625
|
Duplicate vnet.py for future modification
To incorporate styled convolution from style.py
|
2021-03-24 12:36:06 -04:00 |
|
Yin Li
|
a697845933
|
Add style components from stylegan2
* styled convolution with mod and demod
* pixelnorm
* linear and conv layers with equalized learning rate
|
2021-03-24 12:36:06 -04:00 |
|
Yin Li
|
4706a902f6
|
Add parameter reading
|
2021-03-24 12:36:06 -04:00 |
|
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 |
|