fit_generator не будет работать с аргументом batch_size или без него

Question

Я использую кераты для классификации изображений. Когда дело доходит до создания моего fit_generator, у меня есть код:

model.fit_generator(training_data, epochs=30, batch_size=32, validation_data = test_data)

, который дает мне ошибку:

model.fit_generator(training_data, epochs=30, batch_size=32, validation_data = test_data)
TypeError: fit_generator() got an unexpected keyword argument 'batch_size'

, но когда я удаляю аргумент, я получаю ошибку:

raise ValueError('You must specify `batch_size`')
ValueError: You must specify `batch_size`

Кто-нибудь имел эту проблему с fit_generator или может посмотреть, есть ли что-то, что я мог бы пропустить?

Answer 1

Вы должны передавать данные в форме генератора, когда используете model.fit_generator вместо массива. В этом случае кажется, что training_data является массивом. Следовательно, вы не можете указать batch_size. model.fit_generator не принимает batch_size в качестве аргумента. Однако вы можете использовать размер пакета в аргументе steps_per_epoch.