From e5d1ebafba3acc343d03e4128cf4d47cbf22704b Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 16:11:35 -0600 Subject: [PATCH] update --- generator/ctrl/training_utils/training.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/generator/ctrl/training_utils/training.py b/generator/ctrl/training_utils/training.py index 0bb036f..da17aba 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -78,7 +78,6 @@ def input_fn(params=None): return train_data - # the dimension of the transformer embedding_dim = 1280 @@ -166,7 +165,9 @@ 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) -tf.global_variables_initializer() -estimator_model = tf.keras.estimator.model_to_estimator(keras_model=model, config=run_config) +model.fit(input_fn(), epochs=1) -estimator_model.train(input_fn=input_fn, steps=args.iterations) \ No newline at end of file + +# estimator_model = tf.keras.estimator.model_to_estimator(keras_model=model, config=run_config) +# +# estimator_model.train(input_fn=input_fn, steps=args.iterations) \ No newline at end of file