update

2026-09-09 11:13:26 +08:00 · 2019-10-22 16:13:46 -06:00
parent e5d1ebafba
commit 9939bea3f8
1 changed files with 2 additions and 1 deletions
@@ -165,7 +165,8 @@ run_config = tf.contrib.tpu.RunConfig(
                                        input_partition_dims=[[1, 1], [1, 1]], per_host_input_for_training=3))
 tf.logging.set_verbosity(tf.logging.INFO)

-model.fit(input_fn(), epochs=1)
+params = {"batch_size": 2}
+model.fit(input_fn(params=params), epochs=1)


 # estimator_model = tf.keras.estimator.model_to_estimator(keras_model=model, config=run_config)