Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Allow use of a subset of lens/source/psf models #3

Merged
merged 58 commits into from
Mar 2, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
58 commits
Select commit Hold shift + click to select a range
23a4050
Allow use of a subset of lens/source/psf models
JelleAalbers Oct 5, 2022
03ac3c7
modified image simulation to seperate los, main deflector, and subhal…
swagnercarena Oct 27, 2022
ca77330
most of the subhalo functions implemented. Need to write a few simple…
swagnercarena Nov 11, 2022
bba4f1f
added remaining functions to translate to lenstronomy and return thos…
swagnercarena Nov 11, 2022
03948a4
all of the subhalo functions are now written, just need some testing
swagnercarena Nov 11, 2022
7ba3de9
first test done
swagnercarena Nov 11, 2022
e0d4568
finished testing on new cosmology utils
swagnercarena Nov 15, 2022
712ba7d
testing for everything except subhalos
swagnercarena Nov 17, 2022
8853644
more tests
swagnercarena Nov 18, 2022
862d900
one more test
swagnercarena Nov 18, 2022
83dd259
los functions implemented to replace paltas. Still needs testing
swagnercarena Dec 2, 2022
d2b51b5
small bug fixed
swagnercarena Dec 2, 2022
6760ac7
added temporary halo bias functions
swagnercarena Dec 5, 2022
f08fe75
returned the wrong value
swagnercarena Dec 5, 2022
870c8df
fixed a number of big bugs in the los implementation. Still needs a l…
swagnercarena Dec 8, 2022
464c498
implemented lookup table for los for much faster los draws
swagnercarena Dec 9, 2022
da1945c
basic training code that still needs to be modified
swagnercarena Jan 15, 2023
ad8d9b6
reintroduce normalization
swagnercarena Jan 28, 2023
f7bea29
fixed flags
swagnercarena Jan 30, 2023
7a5059d
str not int flag
swagnercarena Jan 30, 2023
9dc1997
main arg
swagnercarena Jan 30, 2023
e2f828e
fixed manual normalization, modified the config a bit, and made sure …
swagnercarena Jan 30, 2023
7e90252
added code for training with a finite image set
swagnercarena Jan 30, 2023
4cde5fb
typo
swagnercarena Jan 30, 2023
67b2a40
one more rng needed to be switched
swagnercarena Jan 30, 2023
cc0a642
need to split rng_images by number of devices
swagnercarena Jan 30, 2023
73f0cbc
bug squashing round 1. For now just wrote tests for remaining subhalo…
swagnercarena Feb 1, 2023
3bf8cf3
more tests of the training pipeline. So far no obvious bugs, although…
swagnercarena Feb 2, 2023
1a9367a
finally found a major bug. Fixed it and wrote tests to catch it in th…
swagnercarena Feb 5, 2023
9b98db0
FLAGS only in main
swagnercarena Feb 6, 2023
356082f
streamlined test to get rid of train_finite
swagnercarena Feb 6, 2023
5651c1f
switch to adam since we're doing a fixed learning rate
swagnercarena Feb 6, 2023
aef1ca9
clean up how train state is done
swagnercarena Feb 6, 2023
432cb31
switch to cosine learning rate decay to see if that improves learning
swagnercarena Feb 9, 2023
c8385eb
switch to steps in config
swagnercarena Feb 9, 2023
8d3886e
adding an input pipeline to read paltas data
swagnercarena Feb 11, 2023
00e7c0e
added the ability to pull from the paltas data.
swagnercarena Feb 12, 2023
2c52fca
added a fix for lack of pmap
swagnercarena Feb 12, 2023
25dc8c8
bug fix
swagnercarena Feb 12, 2023
d0070a6
another bug with dimension
swagnercarena Feb 12, 2023
9206279
turn off los to see if that helps
swagnercarena Feb 12, 2023
1d762b4
cap batch size when using the paltas images
swagnercarena Feb 12, 2023
4dd51ad
missed a not
swagnercarena Feb 12, 2023
d7cad14
add theta_e to training and make sure we perform well on it
Feb 16, 2023
0b75cec
updated code with tests for theta_e
Feb 16, 2023
69f55cb
fundamental rework to how the input_pipeline works to allow for a fle…
Feb 23, 2023
80290fa
fixd a conceptual issue in how the configuration version of the cosmo…
Feb 23, 2023
1512843
made some more tweaks to the config file to make sure everything was …
Feb 23, 2023
1bebfdf
train should work with new input pipeline now
Feb 23, 2023
edf52a0
whitespace fixes, improvments to configs, set model dimensions based …
Feb 24, 2023
0239c28
changed the image simulation to group the subhalo and main deflector …
Feb 25, 2023
bf76bc1
make sure the train script still works with finite rng iteration
Feb 25, 2023
4bfc575
small bug
Feb 25, 2023
3d00850
wrong index for the principal
Feb 25, 2023
0114900
modifications to the configs
Feb 25, 2023
c3d6c5e
get rid of use jaxstronomy flags
Feb 25, 2023
75c1f3c
smaller batch size because of the memory required by all the substruc…
Feb 25, 2023
e268647
Added notebook for making comparisons between paltas and jaxstronomy.…
Mar 2, 2023
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions jaxstronomy/__init__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from . import cosmology_utils
from . import image_simulation
from . import lens_models
from . import los
from . import power_spectrum
from . import psf_models
from . import source_models
Expand Down
Loading