From 0576ec4594361a35d7250abc66305074f00b536f Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 14:11:46 -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 8c87c50..8134076 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -152,7 +152,7 @@ run_config = tf.contrib.tpu.RunConfig( # this step is critical # remember to patch the TF 1.14 file before running the code, else you're going to see errors here -run_config = tf.contrib.tpu.RunConfig( +run_config = tf.estimator.RunConfig( model_dir=args.model_dir, session_config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True)) tf.logging.set_verbosity(tf.logging.INFO)