Tips for tensorflow

posted Mar 9, 2020, 8:30 AM by Teng-Yok Lee [ updated Jan 2, 2021, 7:19 PM ]

This is my note for various tips about tensorflow.

Contents

How lo list all variables in a checkpoint?

Use:

tf.train.list_variables().

REF: https://stackoverflow.com/questions/38944238/how-do-i-list-certain-variables-in-the-checkpoint

Use:

[n.name for n in tf.get_default_graph().as_graph_def().node]

REF: https://stackoverflow.com/questions/36883949/in-tensorflow-get-the-names-of-all-the-tensors-in-a-graph

Comments