Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
253 Commits 2 Branches 1 Tag 456 KiB
1181ff0f63
Commit Graph

209 Commits

Author SHA1 Message Date
Yin Li
1181ff0f63 Fix bug when dis is Tensor 2021-05-20 18:14:40 -04:00
Yin Li
6a3990f90c Fix bug with shared displacement field 2021-05-20 15:17:18 -04:00
Yin Li
b7a97bbcdc Fix slice figures with many zeros 
the bug was triggered when many pixels are zero valued:
taking log of 0 in quantize()
 2021-05-19 17:34:06 -04:00
Yin Li
eccbd68f03 Set matplotlib usetex to False 2021-05-18 14:34:20 -04:00
Yin Li
3d271e9c44 Add Eulerian shape logging 2021-05-18 14:16:54 -04:00
Yin Li
6b5530256b Add scale factor and pad to lag2eul Eulerian field 2021-05-18 13:32:57 -04:00
Yin Li
04d0bea17e Add misc kwargs passing to custom models and norms 2021-05-18 13:32:54 -04: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
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