Merge pull request yandexdataschool#58 from NickVeld/patch-14

justheuristic · web-flow · commit a625531cd74b · 2019-11-27T02:43:39.000+03:00
week8:autoencoders_torch let use acceptable Modules consisting of som…
diff --git a/week08_autoencoders/autoencoders_torch.ipynb b/week08_autoencoders/autoencoders_torch.ipynb
@@ -531,7 +531,7 @@
     "    assert encoder_out.shape[1:]==torch.Size([code_size]),\"encoder must output a code of required size\"\n",
     "    assert decoder_out.shape[1:]==img_shape,   \"decoder must output an image of valid shape\"\n",
     "\n",
-    "    assert (len(list(model.dec.children())) >= 6),  \"decoder must contain at least 3 dense layers\"\n",
+    "    assert (sum([1 + len(list(c1level.children())) for c1level in model.dec.children()]) >= 6),  \"decoder must contain at least 3 dense layers\"\n",
     "\n",
     "    del encoder_out\n",
     "    del decoder_out\n",
