@@ -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": [] | |||
} | |||
] | |||
} | |||
} |