Created using Colaboratory

ultralytics · Dec 27, 2020 · 73cf75f · 73cf75f
1 parent b096221
commit 73cf75f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tutorial.ipynb b/tutorial.ipynb
@@ -1151,11 +1151,11 @@
         "id": "mcKoSIK2WSzj"
       },
       "source": [
-        "# Reproduce results\n",
+        "# Reproduce\n",
         "%%shell\n",
         "for x in yolov5s yolov5m yolov5l yolov5x; do\n",
-        "  python test.py --data coco.yaml --img 640 --conf 0.25 --iou 0.45  # speed\n",
-        "  python test.py --data coco.yaml --img 640 --conf 0.001 --iou 0.65  # mAP\n",
+        "  python test.py --weights $x.pt --data coco.yaml --img 640 --conf 0.25 --iou 0.45  # speed\n",
+        "  python test.py --weights $x.pt --data coco.yaml --img 640 --conf 0.001 --iou 0.65  # mAP\n",
         "done"
       ],
       "execution_count": null,