Каковы различия между параметром batch в yolov3.cfg и параметром batch_size в keras.fit ()? - PullRequest
0 голосов
/ 13 апреля 2020

В чем разница между параметром batch в файле yolov3.cfg и параметром batch_size в keras.fit()? Как я должен установить их? пожалуйста. * * 1005

Ответы [ 2 ]

0 голосов
/ 13 апреля 2020

Размер партии - это термин, используемый в машинном обучении, и относится к числу примеров обучения, использованных в одной итерации.

Традиционно, batch_size выбирается как степень 2 -> 8, 16, 32, 64. . Чем больше размер партии, тем быстрее сходимость.

0 голосов
/ 13 апреля 2020

Разницы нет, размер партии означает, сколько изображений (образцов) будет в мини-партии во время тренировки. Для yolo, обычно в случае логического вывода, batch_size равен 1.

Как бы вы его установили? Go так высоко, как только можете, если у вас не хватает памяти GPU.

...