Skip to content

Commit 976393b

Browse files
authored
Merge pull request #1533 from AustinRochford/advi-init-remove-sample_vp-progress
Remove sample_vp progressbar for ADVI initialization
2 parents d57000d + 71be307 commit 976393b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc3/sampling.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -430,7 +430,7 @@ def init_nuts(init='advi', n_init=500000, model=None):
430430

431431
if init == 'advi':
432432
v_params = pm.variational.advi(n=n_init)
433-
start = pm.variational.sample_vp(v_params, 1)[0]
433+
start = pm.variational.sample_vp(v_params, 1, progressbar=False)[0]
434434
cov = np.power(model.dict_to_array(v_params.stds), 2)
435435
elif init == 'advi_map':
436436
start = pm.find_MAP()

0 commit comments

Comments
 (0)