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

imaging_resolve_jax: use vmap JAXBind bug is fixed

parent aaa56ce4
No related branches found
No related tags found
No related merge requests found
Pipeline #202538 passed
......@@ -100,5 +100,4 @@ samples, state = jft.optimize_kl(
odir=f"results_nifty_re_response_{response}",
resume=False,
callback=callback,
kl_map=jft.smap,
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment