Skip to content

Instantly share code, notes, and snippets.

@tigershen23

tigershen23/pyro_model_guide_7.py

Last active June 9, 2018 20:30
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save tigershen23/dd27813bdd53a2835bcb876c0ae0b6b2 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save tigershen23/dd27813bdd53a2835bcb876c0ae0b6b2 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pyro_model_guide_7.py
adam_params = {"lr": 0.0005}
optimizer = Adam(adam_params)
svi = SVI(model, guide, optimizer, loss=Trace_ELBO())
n_steps = 2501
for step in range(n_steps):
svi.step(data)
if step % 100 == 0:
print_progress()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment