From d55b87d84bc4bf680a0b6dbf20907eb05c7d19ea Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 12 Jan 2020 03:38:27 +0100 Subject: new checkpoint reader --- cli/app/utils/tf_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'cli/app/utils/tf_utils.py') diff --git a/cli/app/utils/tf_utils.py b/cli/app/utils/tf_utils.py index 69a38e4..0453644 100644 --- a/cli/app/utils/tf_utils.py +++ b/cli/app/utils/tf_utils.py @@ -1,4 +1,4 @@ -from tensorflow.python.training import py_checkpoint_reader +from tensorflow.train import py_checkpoint_reader def read_checkpoint(fp_ckpt, key): reader = py_checkpoint_reader.NewCheckpointReader(fp_ckpt) -- cgit v1.2.3-70-g09d2