Commit Graph

263 Commits

Author SHA1 Message Date
534d177a52 Update links 2024-05-17 09:37:34 +02:00
90a14d56b9 Basic cleanup 2024-04-03 09:44:01 +02:00
68e466c983 Remove unnecessary setup.py now 2024-04-02 17:10:08 +02:00
7b32bc8b29 Removed spurious tool 2024-04-02 16:30:41 +02:00
050b6be80c Add pyproject, better cross-referencing 2024-04-02 16:30:23 +02:00
e24343e63d Import patches from Deaglan fork
commit 15e35b2c91d7b4c0a0158747c8059e756d110fad
Author: Deaglan Bartlett <deaglan.bartlett@physics.ox.ac.uk>
Date:   Mon Sep 11 22:51:30 2023 +0200

    Fix power issue in plotting

commit 048e53bb447faa569dfc86a0b29dd1216974fc21
Author: Deaglan Bartlett <deaglan.bartlett@physics.ox.ac.uk>
Date:   Mon Sep 11 20:23:16 2023 +0200

    Fix tgt norm correction bug

commit 088fda3c4c799f3b6d1233ba7419201c78282483
Author: Deaglan Bartlett <deaglan.bartlett@physics.ox.ac.uk>
Date:   Tue Sep 5 15:38:21 2023 +0200

    Minor changes for coca
2024-04-02 16:19:35 +02:00
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