From ccc04408aa9b3e1783e33ad0577a9136e87ae8a0 Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 16:00:10 -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 d0ba983..0bb036f 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -152,7 +152,6 @@ print(model.summary()) # this is where the saved model is presented to the code # the model directory should have the model checkpoint and # a checkpoint file -tf.global_variables_initializer() run_config = tf.contrib.tpu.RunConfig( model_dir=args.model_dir) @@ -167,6 +166,7 @@ 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) estimator_model.train(input_fn=input_fn, steps=args.iterations) \ No newline at end of file