From ed80aec4d5aa188f671ce5c042c4ee78d0d4cf6c Mon Sep 17 00:00:00 2001
From: Shawn Morel <shawn@instance.bio>
Date: Thu, 17 Aug 2023 15:09:22 -0700
Subject: [PATCH] SVI 2 - missing lambda param

I was a little confused about the explanation of handling global variables in the SVI 2 tutorial. Should lambda_I be defined similar to what's proposed in this PR?
---
 examples/svi_part_ii.ipynb | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/svi_part_ii.ipynb b/examples/svi_part_ii.ipynb
index 68149e4..b6f30f5 100644
--- a/examples/svi_part_ii.ipynb
+++ b/examples/svi_part_ii.ipynb
@@ -165,6 +165,7 @@
     "    beta = pyro.sample(\"beta\", ...) # sample the global RV\n",
     "    for i in pyro.plate(\"locals\", len(data), subsample_size=5):\n",
     "        # sample the local RVs\n",
+    "        lambda_i = pyro.param(\"lambda_{}\".format(i), ...)\n",
     "        pyro.sample(\"z_{}\".format(i), ..., lambda_i)\n",
     "```\n",
     "\n",