From 6510c44ba1ba3ce70f9338d6d98674d4e04137fc Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 16:15:02 -0600 Subject: [PATCH] update --- generator/ctrl/training_utils/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/ctrl/training_utils/training.py b/generator/ctrl/training_utils/training.py index 6df42f0..bfb8b0c 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -166,7 +166,7 @@ run_config = tf.contrib.tpu.RunConfig( tf.logging.set_verbosity(tf.logging.INFO) params = {"batch_size": 2} -model.fit(input_fn(params=params), epochs=1) +model.fit(input_fn(params=params), steps_per_epoch=1000, epochs=1) # estimator_model = tf.keras.estimator.model_to_estimator(keras_model=model, config=run_config)