Skip to content
Snippets Groups Projects
Commit 8c8811c7 authored by Jakob Roth's avatar Jakob Roth
Browse files

imaging_resolve_jax: do not use vmap for now. 

See: https://github.com/NIFTy-PPL/JAXbind/issues/16
parent 9b8403ff
Branches
Tags
No related merge requests found
Pipeline #202214 passed
Stage: build_docker
Stage: testing
Stage: release
Stage: deploy
Show whitespace changes
Inline Side-by-side
demo/imaging_resolve_jax.py
+ 1
− 0
View file @ 8c8811c7
......@@ -106,4 +106,5 @@ samples, state = jft.optimize_kl(
odir=f"results_nifty_re_response_{response}",
resume=False,
callback=callback,
kl_map=jft.smap,
)
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment