Update tutorial.ipynb (#7212)

2022-03-30 12:53:49 +02:00 · 2022-03-30 12:53:49 +02:00 · 9c1e91aea2
parent cf4f3c3455
commit 9c1e91aea2
1 changed files with 1 additions and 1 deletions
--- a/tutorial.ipynb
+++ b/tutorial.ipynb
@ -1077,7 +1077,7 @@
      },
      "source": [
        "# VOC\n",
-        "for b, m in zip([64, 64, 32, 16], ['yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']):  # zip(batch_size, model)\n",
+        "for b, m in zip([64, 64, 64, 32, 16], ['yolov5n', 'yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']):  # zip(batch_size, model)\n",
        "  !python train.py --batch {b} --weights {m}.pt --data VOC.yaml --epochs 50 --cache --img 512 --nosave --hyp hyp.VOC.yaml --project VOC --name {m}"
      ],
      "execution_count": null,