diff options
author | 2018-04-20 12:40:57 -0700 | |
---|---|---|
committer | 2018-04-20 12:43:51 -0700 | |
commit | 517d1912f4ec71180944320350a3694332a1dedc (patch) | |
tree | 23fd05eb7e72c8ce13766a516280622089cd0c70 /tensorflow/contrib/BUILD | |
parent | b23e91d247368f2046dae035b5c7bdda56512077 (diff) |
Add a utility to visualize object-based checkpoints
Useful for generating a warm fuzzy feeling that everything you think should be saved was saved, and for explaining what object-based checkpointing is. (Also useful on the former front will be a planned "assert that all of this Graph's trainable variables are accessible from object X" function.)
Somewhat hacky since it generates strings rather than using the pydot bindings (and so works without a pydot dependency).
PiperOrigin-RevId: 193708003
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r-- | tensorflow/contrib/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index 7e47516550..d28392a62c 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -25,6 +25,7 @@ py_library( "//tensorflow/contrib/batching:batch_py", "//tensorflow/contrib/bayesflow:bayesflow_py", "//tensorflow/contrib/boosted_trees:init_py", + "//tensorflow/contrib/checkpoint/python:checkpoint", "//tensorflow/contrib/cloud:cloud_py", "//tensorflow/contrib/cluster_resolver:cluster_resolver_pip", "//tensorflow/contrib/cluster_resolver:cluster_resolver_py", |