Commit graph

127 commits

Author SHA1 Message Date
7361c3eb9d chore: Blackify the source code for pretty formatting 2024-05-20 17:50:32 +02:00
90a14d56b9 Basic cleanup 2024-04-03 09:44:01 +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
e20a3e3f62 Cleanup, Enable plt_power by default 2021-05-30 22:58:33 -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
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
abb16fe26a Improve FieldDataset.assemble 2021-03-26 12:47:38 -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
8544ff07e8 Fixes and cleaning up 2021-03-18 14:24:44 -04:00
Yin Li
156011be5f Fix indentation bug pointed out by Patrick Lachance 2020-09-21 19:56:50 -04:00
Yin Li
b7a9adf0c7 Add pytorch version reporting 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
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
f3216f4cbd Add multiple pkgs to import_attr 2020-09-09 14:55:36 -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
5d22594ede Add shape report after narrow_cast in training 2020-08-22 12:28:08 -07:00
Yin Li
9c8b331bf5 Add loglxe loss as log of product of Lagrangian and Eulerian losses 2020-08-09 12:59:24 -07:00
Yin Li
265587922d Add srsgan model and scale_factor to model arguments 2020-08-04 21:13:10 -07:00
Yin Li
b54fc4ba3a Add a line about socket hopefully will avoid port conflicts 2020-07-28 07:38:07 -07:00
Yin Li
a886e53c54 Always use training loss for lr scheduler 2020-07-27 17:47:58 -07:00
Yin Li
2336d83f2d Add optimizer and scheduler kwargs via json.loads 2020-07-27 12:40:19 -07:00
Yin Li
8ce13e67f6 Fix training hang due to constrained layout of matplotlib 2020-07-27 12:39:41 -07:00
Yin Li
ca39b2028f Replace cuda.device with env var to assign GPU to processes 2020-07-22 18:48:12 -07:00
Yin Li
98cdd4795c Add DistFieldSampler to benefit from page cache 2020-07-21 23:37:59 -07:00
Yin Li
ec46f41ba5 Add lag x eul loss tracking to tensorboard 2020-07-21 21:10:34 -07:00
Yin Li
56f8fa932b Revert "Deprecate GroupedRandomSampler for use of memmap in place of cache"
This reverts commit bbf77c9f91.
2020-07-16 15:22:52 -07:00
Yin Li
4160ed25f7 Add lagrangian and eulerian alternate training 2020-07-15 15:57:10 -04:00
Yin Li
e8744d6c7b Remove adversary 2020-07-14 21:07:05 -04:00
Yin Li
6fa85f8285 Merge branch 'master' into lag2eul 2020-07-14 18:47:54 -04:00
Yin Li
bbf77c9f91 Deprecate GroupedRandomSampler for use of memmap in place of cache 2020-07-14 18:47:26 -04:00
Yin Li
3437b20ed8 Merge branch 'master' into lag2eul 2020-07-14 18:28:10 -04:00
Yin Li
818ed6923d Add memmap to numpy data loading 2020-07-14 17:42:10 -04:00
Yin Li
d0ab596e11 Remove min-reduction; Split grad plots; Disable cgan w/o adv 2020-07-13 09:40:15 -04:00
Yin Li
29ab550032 Change instance noise
Simpler linear annealing as in the blog post;
different noise to output and target
2020-07-11 01:56:46 -04:00
Yin Li
873258d8a7 Add linear resamplers 2020-07-11 01:37:23 -04:00
Yin Li
c956dca795 Add narrow_cast and narrow_by 2020-07-11 01:24:20 -04:00
Yin Li
28ec245a7a Add pixel-shifting data augmentation 2020-07-11 00:46:19 -04:00
Yin Li
8a95d69818 Add cropping anchors controlled by start, stop, step 2020-07-11 00:39:19 -04:00
Yin Li
819e77cd86 Add slice plotting for field >2d 2020-07-10 14:58:08 -04:00
Yin Li
bab3f08aa5 Add Lag2Eul to training 2020-07-10 12:50:07 -04:00