From b951342c17603c7f0d6d726386d6e98506612bd2 Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 14:12:27 -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 8134076..bc2f13c 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -144,7 +144,7 @@ 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 -run_config = tf.contrib.tpu.RunConfig( +run_config = tf.estimator.RunConfig( model_dir=args.model_dir)