experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix example by deleting bad steps_per_epoch value and associated TODO.

Browse Source 
PiperOrigin-RevId: 238426188
This commit is contained in:
A. Unique TensorFlower 2019-03-14 05:55:30 -07:00 committed by TensorFlower Gardener
parent 869788b0e6
commit 66df637441
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/examples/saved_model/integration_tests/use_mnist_cnn.py
View File

@ -103,16 +103,13 @@ def main(argv):
fake_tiny_data=FLAGS.fast_test_mode) fake_tiny_data=FLAGS.fast_test_mode)
model.compile(loss=tf.keras.losses.categorical_crossentropy, model.compile(loss=tf.keras.losses.categorical_crossentropy,
optimizer=tf.keras.optimizers.SGD(), optimizer=tf.keras.optimizers.SGD(),
metrics=['accuracy'], metrics=['accuracy'])
# TODO(arnoegw): Remove after investigating huge allocs.
run_eagerly=True)
print('Training on %s with %d trainable and %d untrainable variables.' % print('Training on %s with %d trainable and %d untrainable variables.' %
('Fashion MNIST' if FLAGS.use_fashion_mnist else 'MNIST', ('Fashion MNIST' if FLAGS.use_fashion_mnist else 'MNIST',
len(model.trainable_variables), len(model.non_trainable_variables))) len(model.trainable_variables), len(model.non_trainable_variables)))
model.fit(x_train, y_train, model.fit(x_train, y_train,
batch_size=128, batch_size=128,
epochs=FLAGS.epochs, epochs=FLAGS.epochs,
steps_per_epoch=3,
verbose=1, verbose=1,
validation_data=(x_test, y_test)) validation_data=(x_test, y_test))