update

2026-09-10 11:40:48 +08:00 · 2019-10-22 14:12:27 -06:00
parent 0576ec4594
commit b951342c17
1 changed files with 1 additions and 1 deletions
@@ -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)