Skip to content

Instantly share code, notes, and snippets.

@thomwolf

thomwolf/read_checkpoint.py

Created August 8, 2019 17:15
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 thomwolf/6759298850357ddf8a153bcce6f049cc to your computer and use it in GitHub Desktop.
Save thomwolf/6759298850357ddf8a153bcce6f049cc to your computer and use it in GitHub Desktop.
Download ZIP
Read a TensorFlow checkpoint
Raw
read_checkpoint.py
import os
from pprint import pprint
import tensorflow as tf
tf_path = os.path.abspath('./models/117M/model.ckpt') # Path to our TensorFlow checkpoint
tf_vars = tf.train.list_variables(tf_path)
pprint(tf_vars)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment