КАК (как) Keras и Pytorch обрабатывают последний пакет, когда batch_size не кратен размеру обучающих данных? - PullRequest
0 голосов
/ 15 апреля 2020

Прежде чем вы предложите или сообщите как дубликат, я спрашиваю КАК , решают ли две библиотеки проблему НЕ ТО, ЧТО ПРОИСХОДИТ , потому что я знаю, что она может создать пакет из оставшихся данных , И я знаю, что он справляется, но все, что я прошу, это КАК .

Если у меня есть 100 изображений в качестве обучающих данных и batch_size=15, в последней партии будет 10 изображений для обучения. Мой вопрос заключается в том, что когда слой Input() уже знает, что данные поступают в виде (Batch_size,channel,width,height) для PyTorch и (batch_size,width,height,channels) для Keras, использующих Tensorflow в качестве бэкэнда.

Если последняя партия имеет размер = 10, разве модель не должна выдавать ошибку, потому что вместо (15,1,28,28) она получит (10,1,28,28), учитывая, что у нас (28,28) пикселей в градациях серого?

Что происходит за кулисами?

1 Ответ

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

Например, если вы посмотрите на do c, вы увидите, что размер партии является необязательным. Это потому, что размер пакета обрабатывается как переменная на любой итерации

...