web-dev-qa-db-fra.com

Comment inspecter des variables dans un fichier de point de contrôle dans TensorFlow lorsque TensorFlow ne peut pas trouver l'attribut tools?

J'essayais d'inspecter les points de contrôle en utilisant le code situé à inspect_checkpoint.py . Cependant, je n'ai pas pu le faire fonctionner car ils ne donnaient pas vraiment d'exemple. J'ai essayé la chose la plus simple que je pensais qui fonctionnerait:

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

cependant, je reçois que python n'a pas d'attribut tools:

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

cela ressemble à un bug/problème (gênant) trivial. Est-ce que quelqu'un sait ce qui se passe? Pourquoi ne trouve-t-il pas d'outils? En outre, même s'il le trouvait, comment exécuter la fonction fournie dans ce fichier?


Malheureusement, la question elle-même n’a pas vraiment permis de résoudre le problème. La question est ici Comment trouver les noms de variables sauvegardées dans le point de contrôle tensorflow?

7
Pinocchio

Essaye ça:

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)

L'argument all_tensors a été ajouté depuis Tensorflow 0.12.0-rc0 .

14
sagunms

Inspect_checkpoint.py n'est-il pas un binaire?

Quelque chose comme ça pourrait marcher:

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

MODIFIER:

Ou sans bazel:

Recherchez où tensorflow est installé et exécutez la commande avec python:

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

Pour toutes les options, voir le fichier lui-même: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/inspect_checkpoint.py

2
guinny

Vous pouvez également utiliser l'interface de ligne de commande qui utilise inspect_checkpoint.

0
drevicko