Update captioning_torch.ipynb

justheuristic · web-flow · commit 980121c7b314 · 2018-10-29T20:31:50.000+03:00
diff --git a/week07_seq2seq/captioning_torch.ipynb b/week07_seq2seq/captioning_torch.ipynb
@@ -382,7 +382,7 @@
    "source": [
     "dummy_loss = compute_loss(network, dummy_img_vec, dummy_capt_ix)\n",
     "\n",
-    "assert dummy_loss.shape == torch.Size([1]), 'loss must be scalar'\n",
+    "assert len(dummy_loss.shape) <= 1, 'loss must be scalar'\n",
     "assert dummy_loss.data.numpy() > 0, \"did you forget the 'negative' part of negative log-likelihood\"\n",
     "\n",
     "dummy_loss.backward()\n",
