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?
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 .
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