Merge pull request #492 from shreyas-bk/master

use latest saved weights in tutorial

Author: zylo117

tutorial/train_shape.ipynb

@@ -463,7 +463,15 @@
         "outputId": "f38ef0e7-bf30-428a-cf93-43e43a60fdae"
       },
       "source": [
-        "! python coco_eval.py -c 0 -p shape -w logs/shape/efficientdet-d0_49_1400.pth"
+        "#get latest weight file\n",
+        "%cd logs/shape\n",
+        "weight_file = !ls -Art | grep efficientdet\n",
+        "%cd ../..\n",
+        "\n",
+        "#uncomment the next line to specify a weight file\n",
+        "#weight_file[-1] = 'efficientdet-d0_49_1400.pth'\n",
+        "\n",
+        "! python coco_eval.py -c 0 -p shape -w \"logs/shape/{weight_file[-1]}\""
       ],
       "execution_count": 4,
       "outputs": [
@@ -575,7 +583,7 @@
         "                             ratios=[(1.0, 1.0), (1.4, 0.7), (0.7, 1.4)],\n",
         "                             scales=[2 ** 0, 2 ** (1.0 / 3.0), 2 ** (2.0 / 3.0)])\n",
         "\n",
-        "model.load_state_dict(torch.load('logs/shape/efficientdet-d0_49_1400.pth'))\n",
+        "model.load_state_dict(torch.load('logs/shape/'+weight_file[-1]))\n",
         "model.requires_grad_(False)\n",
         "model.eval()\n",
         "\n",