Move tensor to the CPU for t-SNE fitting (#25)

Atcold · Jul 15, 2021 · b28aa29 · b28aa29
1 parent 42c0c5f
commit b28aa29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/11-VAE.ipynb b/11-VAE.ipynb
@@ -325,7 +325,7 @@
     "epochs = (0, 5, 10)\n",
     "for epoch in epochs:\n",
     "    Y.append(codes['μ'][epoch][:N])\n",
-    "    E.append(TSNE(n_components=2).fit_transform(Y[-1]))\n",
+    "    E.append(TSNE(n_components=2).fit_transform(Y[-1].detach().cpu()))\n",
     "    X.append(codes['x'][epoch][:N])"
    ]
   },