Fix warning in gmm tutorial
"\n",
"# Register hooks to monitor gradient norms.\n",
"gradient_norms = defaultdict(list)\n",
- "inference.loss(model, guide, data) # Initializes param store.\n",
+ "inference.loss(model, config_enumerate(guide, 'parallel'), data) # Initializes param store.\n",
"for name, value in pyro.get_param_store().named_parameters():\n",
" value.register_hook(lambda g, name=name: gradient_norms[name].append(g.norm().item()))\n",
"\n",
