This commit is contained in:
nickwalton
2019-10-22 15:54:58 -06:00
parent ae3f3b84af
commit ecb73576c0
@@ -152,6 +152,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
tf.global_variables_initializer()
run_config = tf.contrib.tpu.RunConfig(
model_dir=args.model_dir)