], | ], | ||||
"execution_count": null, | "execution_count": null, | ||||
"outputs": [] | "outputs": [] | ||||
}, | |||||
{ | |||||
"cell_type": "code", | |||||
"metadata": { | |||||
"id": "BSgFCAcMbk1R" | |||||
}, | |||||
"source": [ | |||||
"# VOC\n", | |||||
"for b, m in zip([64, 48, 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, | |||||
"outputs": [] | |||||
} | } | ||||
] | ] | ||||
} | } |