2017-03-22 3 views
-3

Un modèle est entraîné avec GPU et le résultat est sauvegardé par fichier de point de contrôle. Le fichier de point de contrôle enregistré peut être exécuté par cpu-tensorflow? Si non, peut convertir le fichier de point de contrôle enregistré afin d'exécuter le modèle dans cpu-tensorflow?Tensorflow: Modèle entraîné (fichiers de point de contrôle) sur GPU peut être converti en modèle exécutant CPU?

+7

Copie possible de [Un modèle peut-il être formé sur un CPU utilisé sur CPU pour l'inférence et vice versa?] (Http://stackoverflow.com/questions/40980035/can-a-model-trained-on-gpu-us- on-cpu-for-inference-and-versa) – kaufmanu

Répondre

0

Oui! Normalement, ça peut! L'exception est with tf.device('gpu:0') instructions. Si vous ne les avez pas dans votre code, vous êtes prêt à partir!

Bonne chance!

+1

Si vous avez des affectations de périphériques dans votre code, vous pouvez les supprimer du fichier GraphDef .pb enregistré avec la règle de réécriture remove_device: https://github.com/ tensorflow/tensorflow/arbre/maître/tensorflow/tools/graph_transforms/# remove_device –