diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-12 03:38:27 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-12 03:38:27 +0100 |
| commit | d55b87d84bc4bf680a0b6dbf20907eb05c7d19ea (patch) | |
| tree | 360aa52ac1556724461bea57586e3bd70292caf0 /cli/app | |
| parent | 745ef49ffb5bc2f2eefed349149d7effd1de5f2b (diff) | |
new checkpoint reader
Diffstat (limited to 'cli/app')
| -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) |