From ecb73576c0ecabb7ef4e51f1f817bb8e2045ded7 Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 15:54:58 -0600 Subject: [PATCH] update --- generator/ctrl/training_utils/training.py | 1 + 1 file changed, 1 insertion(+) diff --git a/generator/ctrl/training_utils/training.py b/generator/ctrl/training_utils/training.py index 0cb78a2..d0ba983 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -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)