model.fit_generator(datagen.flow(X_train,y_train,batch_size=32),epochs=10,steps_per_epoch=5000,validation_data=(X_test,y_test))
我的总数据大小为5000,批量大小为32,那么如何确定steps_per_epoch的值
案例1:何时不使用ImageAugumentation
cas2 2:使用ImageAugumentation时(Coz数字图像会增加以及如何在steps_per_epoch中包含它)
答案 0 :(得分:5)
steps_per_epoch
将是训练集中(扩充前)的样本总数除以批量大小。所以 -
steps_per_epoch = 5000 / 32 ~ 156
使用数据扩充不会影响此计算。您还可以在我Training a CNN with Keras的视频中获取有关使用此参数以及fit_generator()
的更多信息。 steps_per_epoch
报道大约在4:08开始。