bugfix for mode = "both"

byronknoll · Sep 7, 2020 · 2d9ee55 · 2d9ee55
1 parent d215bbd
commit 2d9ee55
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tensorflow-compress.ipynb b/tensorflow-compress.ipynb
@@ -7,7 +7,7 @@
       "provenance": [],
       "collapsed_sections": [],
       "toc_visible": true,
-      "authorship_tag": "ABX9TyO7Lt6myOfZDaien4SwLBx0"
+      "authorship_tag": "ABX9TyO9tL2MNcmjnOlaE0tZLeS8"
     },
     "kernelspec": {
       "name": "python3",
@@ -548,7 +548,8 @@
         "    coder.finish()\n",
         "  print(template.format(100, -cross_entropy / length, time.time() - start))\n",
         "  system_info()\n",
-        "  model.save_weights('./data/model')\n"
+        "  if mode != \"both\" or not compress:\n",
+        "    model.save_weights('./data/model')\n"
       ],
       "execution_count": null,
       "outputs": []
@@ -984,7 +985,7 @@
         "  download('data/compressed.dat')\n",
         "  if preprocess == 'nncp':\n",
         "    download('data/dictionary.words')\n",
-        "  if checkpoint:\n",
+        "  if checkpoint and mode != \"both\":\n",
         "    download('data/model.index')\n",
         "    download('data/model.data-00000-of-00001')\n",
         "    download('data/checkpoint')"