Wie untersucht man Variablen in einer Prüfpunktdatei in TensorFlow, wenn TensorFlow das Werkzeugattribut nicht finden kann?

Question

Ich habe versucht, Checkpoints mit dem Code unter inspect_checkpoint.py zu überprüfen. Ich konnte es jedoch nicht funktionieren lassen, weil sie nicht wirklich ein Beispiel lieferten. Ich versuchte, die einfachste Sache, die ich würde funktionieren dachte:

tf.python.tools.inspect_checkpoint.print_tensors_in_checkpoint_file(file_name='./tmp/mdl_ckpt',tensor_name='',all_tensors='') 

aber ich bekommen, dass python kein Attribut hat tools:

AttributeError: module 'tensorflow.python' has no attribute 'tools' 

es wie ein (peinliche) triviale Fehler/Problem scheint. Weiß jemand, was vor sich geht? Warum kann es keine Werkzeuge finden? Selbst wenn es es gefunden hätte, wie würde man dann die Funktion in dieser Datei ausführen?

---

Leider lieferte die sehr verwandte Frage nicht wirklich eine Antwort, wie man dieses Problem umgehen kann. Die Frage ist hier How can find the variable names that saved in tensorflow checkpoint?

Answer 1

Nun, ist inspect_checkpoint.py eine Binärdatei?

So etwas wie dies funktionieren könnte:

bazel run tensorflow/python/tools:inspect_checkpoint -- --file_name=YOUR_CKPT 

EDIT:

oder ohne bazel:

finden, wo tensorflow installiert ist, und führen Sie den Befehl mit python:

python PATH_TO_VENV/lib/python3.6/site-packages/tensorflow/python/tools/inspect_checkpoint.py --file_name=YOUR_CKPT 

Informationen zu allen Optionen finden Sie in der Datei selbst: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/inspect_checkpoint.py

Answer 2

Try this:

from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file 
print_tensors_in_checkpoint_file(file_name='./tmp/mdl_ckpt', tensor_name='', all_tensors=False) 

Das all_tensors Argument wurde seit Tensorflow 0.12.0-rc0 hinzugefügt.