From 68dc181a5e34d1f20edabe531176b3bfb50001f9 Mon Sep 17 00:00:00 2001 From: Abhai Kollara Dilip Date: Tue, 29 Jan 2019 02:54:21 +0530 Subject: [PATCH] Minor docs fix for fit_generator (#12153) * Docs fix * Fix docstring style --- keras/engine/training.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/keras/engine/training.py b/keras/engine/training.py index 85ab5e23e8e..08a172fd8b1 100644 --- a/keras/engine/training.py +++ b/keras/engine/training.py @@ -1347,8 +1347,7 @@ def fit_generator(self, generator, Total number of steps (batches of samples) to yield from `generator` before declaring one epoch finished and starting the next epoch. It should typically - be equal to the number of samples of your dataset - divided by the batch size. + be equal to `ceil(num_samples / batch_size)` Optional for `Sequence`: if unspecified, will use the `len(generator)` as a number of steps. epochs: Integer. Number of epochs to train the model.