Skip to content

Commit

Permalink
Minor docs fix for fit_generator (keras-team#12153)
Browse files Browse the repository at this point in the history
* Docs fix

* Fix docstring style
  • Loading branch information
abhaikollara authored and fchollet committed Jan 28, 2019
1 parent 59fd3ce commit 68dc181
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions keras/engine/training.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit 68dc181

Please sign in to comment.