Skip to content

Commit 89a56e8

Browse files
committed
Train on the whole dataset
1 parent 456b8d7 commit 89a56e8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lstm/imdb-classifier/imdb-cnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@
5555
optimizer='adam',
5656
metrics=['accuracy'])
5757

58-
model.fit(X_train[:1000], y_train[:1000],
58+
model.fit(X_train, y_train,
5959
batch_size=config.batch_size,
6060
epochs=config.epochs,
6161
validation_data=(X_test, y_test), callbacks=[WandbCallback()])

0 commit comments

Comments
 (0)