Update 06_multicat.ipynb (#349)

Fix binary_cross_entropy
2020-12-11 16:36:58 -05:00 · 2020-12-11 16:36:58 -05:00 · 1cf5630742
commit 1cf5630742
parent 08746b427e
1 changed files with 1 additions and 1 deletions
--- a/06_multicat.ipynb
+++ b/06_multicat.ipynb
@ -976,7 +976,7 @@
   "source": [
    "def binary_cross_entropy(inputs, targets):\n",
    "    inputs = inputs.sigmoid()\n",
-    "    return -torch.where(targets==1, inputs, 1-inputs).log().mean()"
+    "    return -torch.where(targets==1, 1-inputs, inputs).log().mean()"
   ]
  },
  {