Эта ошибка вызвана тем, что размер обучающих данных меньше batch_size
, что недопустимо. Значение batch_size по умолчанию - 32, что означает, что количество обучающих изображений должно быть не менее 32.
Для его решения вам нужно выбрать одно из следующих решений.
- Установить
batch_size
меньше, чем размер обучающих данных, например:
image_classifier.create(train_data, batch_size=4)
- Увеличьте размер обучающих данных, добавив больше данных.
Как для len()
, мы его добавили. Вы можете попробовать len(train_data)
снова или использовать train_data.size
.