Skip to content

Commit

Permalink
Update tutorial.ipynb (#7255)
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher authored Apr 3, 2022
1 parent 3d3483c commit 035b554
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tutorial.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -1077,7 +1077,7 @@
},
"source": [
"# VOC\n",
"for b, m in zip([64, 64, 64, 32, 16], ['yolov5n', 'yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']): # zip(batch_size, model)\n",
"for b, m in zip([64, 64, 64, 32, 16], ['yolov5n', 'yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']): # batch, model\n",
" !python train.py --batch {b} --weights {m}.pt --data VOC.yaml --epochs 50 --img 512 --hyp hyp.VOC.yaml --project VOC --name {m} --cache"
],
"execution_count": null,
Expand Down

0 comments on commit 035b554

Please sign in to comment.