Skip to content

Instantly share code, notes, and snippets.

@JasonTam

JasonTam/tf_inspect_checkpoint_example.py

Created May 30, 2018 18:17
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save JasonTam/0dd1c1f679163f98b000d60a264417d4 to your computer and use it in GitHub Desktop.
Save JasonTam/0dd1c1f679163f98b000d60a264417d4 to your computer and use it in GitHub Desktop.
Download ZIP
exmaple of how to inspect the contents of a tensorflow V2 checkpoint
Raw
tf_inspect_checkpoint_example.py
from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file
print_tensors_in_checkpoint_file(
file_name='/tmp/tensorboard-logs/tophat-movielens/20180530T140423/model.ckpt',
tensor_name=None, all_tensors=False,
all_tensor_names=True,
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment