Skip to content

Commit e31a141

Browse files
committed
adding message the performance can be improved with more epochs
1 parent 7cbcdb3 commit e31a141

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

intermediate_source/char_rnn_classification_tutorial.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -337,7 +337,8 @@ def train(rnn, training_data, n_epoch = 10, n_batch_size = 64, report_every = 50
337337
return all_losses
338338

339339
##########################################################################
340-
# We can now train a dataset with minibatches for a specified number of epochs
340+
# We can now train a dataset with minibatches for a specified number of epochs. The number of epochs for this
341+
# example is reduced to speed up the build. You can get better results with different parameters.
341342

342343
start = time.time()
343344
all_losses = train(rnn, train_set, n_epoch=27, learning_rate=0.15, report_every=5)

0 commit comments

Comments
 (0)