Progress bar for fori_loop #13126
Answered
by
sharadmv
rezaaskary
asked this question in
Ideas
-
Hi, does anyone know an efficient way to add progress bar to lax.fori_loop? something like integrating tqdm with for loop. |
Beta Was this translation helpful? Give feedback.
Answered by
sharadmv
Nov 6, 2022
Replies: 1 comment 1 reply
-
I created an example of you how can use https://gist.github.com/sharadmv/be24b3107cf9b8bf027ea8e2f177882e Does this work? |
Beta Was this translation helpful? Give feedback.
1 reply
Answer selected by
rezaaskary
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I created an example of you how can use
jax.debug.callback
to create a progress bar system for JAX:https://gist.github.com/sharadmv/be24b3107cf9b8bf027ea8e2f177882e
Does this work?