Fix enumeration notebook tiny problem
"@config_enumerate\n",
"def model(data, num_components=3):\n",
" print('Running model with {} data points'.format(len(data)))\n",
- " p = pyro.sample(\"p\", dist.Dirichlet(0.5 * torch.ones(3)))\n",
+ " p = pyro.sample(\"p\", dist.Dirichlet(0.5 * torch.ones(num_components)))\n",
" scale = pyro.sample(\"scale\", dist.LogNormal(0, num_components))\n",
" with pyro.plate(\"components\", num_components):\n",
" loc = pyro.sample(\"loc\", dist.Normal(0, 10))\n",
