Merge pull request yandexdataschool#57 from NickVeld/patch-13

justheuristic · web-flow · commit 16454e972682 · 2019-11-27T02:43:25.000+03:00
week8:autoencoders_torch memory releasing after big code_size tests
diff --git a/week08_autoencoders/autoencoders_torch.ipynb b/week08_autoencoders/autoencoders_torch.ipynb
@@ -533,6 +533,9 @@
     "\n",
     "    assert (len(list(model.dec.children())) >= 6),  \"decoder must contain at least 3 dense layers\"\n",
     "\n",
+    "    del encoder_out\n",
+    "    del decoder_out\n",
+    "    torch.cuda.empty_cache()\n",
     "print(\"All tests passed!\")"
    ]
   },
