Commit Graph

246 Commits

Author SHA1 Message Date
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