View source on GitHub |
Make a progress_bar_fn
that uses tqdm
.
tfp.experimental.mcmc.make_tqdm_progress_bar_fn(
description='', leave=True
)
Args | |
---|---|
description
|
str to display next to the progress bar, default is "".
|
leave
|
Boolean whether to leave the progress bar up after finished. |
Returns | |
---|---|
tqdm_progress_bar_fn
|
A function that takes an integer num_steps and
returns a tqdm progress bar iterator.
|