diff options
Diffstat (limited to 'cli/app/utils/tf_utils.py')
| -rw-r--r-- | cli/app/utils/tf_utils.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |