Yin Li
|
352482d475
|
Remove Lagrangian part
|
2021-06-08 18:28:45 -04:00 |
|
Yin Li
|
0d4ae3424e
|
Add cuda backend to inference
|
2021-05-30 23:04:50 -04:00 |
|
Yin Li
|
e20a3e3f62
|
Cleanup, Enable plt_power by default
|
2021-05-30 22:58:33 -04:00 |
|
Yin Li
|
c7836ad7be
|
Cleanup name and comments
|
2021-05-29 17:21:25 -04:00 |
|
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
|
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 |
|