Created using Colaboratory

ultralytics · Jan 19, 2022 · e521909 · e521909
1 parent 0cf932b
commit e521909
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorial.ipynb b/tutorial.ipynb
@@ -1077,7 +1077,7 @@
       },
       "source": [
         "# VOC\n",
-        "for b, m in zip([64, 48, 32, 16], ['yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']):  # zip(batch_size, model)\n",
+        "for b, m in zip([64, 64, 32, 16], ['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.finetune.yaml --project VOC --name {m}"
       ],
       "execution_count": null,
@@ -1099,4 +1099,4 @@
       "outputs": []
     }
   ]
-}
+}