Tutorial: update to differentiate torch.optim.Adam and pyro.optim.Adam
"metadata": {},
"outputs": [],
"source": [
- "from pyro import optim\n",
"from pyro.infer import SVI, Trace_ELBO\n",
"\n",
"\n",
- "adam = optim.Adam({\"lr\": 0.03})\n",
+ "adam = pyro.optim.Adam({\"lr\": 0.03})\n",
"svi = SVI(model, guide, adam, loss=Trace_ELBO())"
]
},
