PyTorch Implementation Added

colors-lab · Oct 19, 2020 · 0c95f39 · 0c95f39
1 parent 0136a4f
commit 0c95f39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CNMP_PyTorch.ipynb b/CNMP_PyTorch.ipynb
@@ -321,7 +321,7 @@
     "        \n",
     "        # Encoder takes observations which are (X,Y) tuples and produces latent representations for each of them\n",
     "        self.encoder = nn.Sequential(\n",
-    "        nn.Linear(d_x+d_x,128),nn.ReLU(),\n",
+    "        nn.Linear(d_x+d_y,128),nn.ReLU(),\n",
     "        nn.Linear(128,128),nn.ReLU(),\n",
     "        nn.Linear(128,128)\n",
     "        )\n",