add grep

Author: shreyas-bk

tutorial/train_shape.ipynb

@@ -465,13 +465,13 @@
       "source": [
         "#get latest weight file\n",
         "%cd logs/shape\n",
-        "weight_file = !ls -Art | tail -n 1\n",
+        "weight_file = !ls -Art | grep efficientdet\n",
         "%cd ../..\n",
         "\n",
         "#uncomment the next line to specify a weight file\n",
-        "#weight_file[0] = 'efficientdet-d0_49_1400.pth'\n",
+        "#weight_file[-1] = 'efficientdet-d0_49_1400.pth'\n",
         "\n",
-        "! python coco_eval.py -c 0 -p shape -w \"logs/shape/{weight_file[0]}\""
+        "! python coco_eval.py -c 0 -p shape -w \"logs/shape/{weight_file[-1]}\""
       ],
       "execution_count": 4,
       "outputs": [
@@ -583,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/'+weight_file[0]))\n",
+        "model.load_state_dict(torch.load('logs/shape/'+weight_file[-1]))\n",
         "model.requires_grad_(False)\n",
         "model.eval()\n",
         "\n",