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?
-3
A
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 –
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