Created using Colaboratory

This commit is contained in:
Glenn Jocher committed 2022-01-19 10:18:29 -10:00
1 parent 0cf932bf63
commit e5219099cd
1 file changed
+1 -1
+1 -1
View File
@@ -1077,7 +1077,7 @@
}, },
"source": [ "source": [
"# VOC\n", "# 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}" " !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, "execution_count": null,