Skip to content

Commit 486b05c

Browse files
authored
Add note about fit_generator (#105)
1 parent 2eb584a commit 486b05c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

README.rst

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,8 @@ Simple training pipeline
9494
model.compile('Adam', loss=bce_jaccard_loss, metrics=[iou_score])
9595
9696
# fit model
97+
# if you use data generator use model.fit_generator(...) instead of model.fit(...)
98+
# more about `fit_generator` here: https://keras.io/models/sequential/#fit_generator
9799
model.fit(
98100
x=x_train,
99101
y=y_train,

0 commit comments

Comments
 (0)