Fix steps_per_epochs

xhwhappy · Apr 1, 2017 · 19babe3 · 19babe3
1 parent f076609
commit 19babe3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cifar10.py b/cifar10.py
@@ -80,7 +80,7 @@
 
     # Fit the model on the batches generated by datagen.flow().
     model.fit_generator(datagen.flow(X_train, Y_train, batch_size=batch_size),
-                        steps_per_epoch=X_train.shape[0],
+                        steps_per_epoch=X_train.shape[0] // batch_size,
                         validation_data=(X_test, Y_test),
                         epochs=nb_epoch, verbose=1, max_q_size=100,
                         callbacks=[lr_reducer, early_stopper, csv_logger])