Ich möchte die Platzierung des Geräts im Tensorboard mit dem folgenden Code untersuchen, um das Diagramm in der Zusammenfassung zu erstellen.
%Vor%Dies funktioniert zum Anzeigen des Diagramms und der Zusammenfassungen, die in der Grafik definiert sind. Bei der Auswahl der 'Geräteplatzierung' im Tensorboard werden jedoch alle Knoten 'unbekanntes Gerät' zugewiesen. Muss ich die Geräteplatzierung auf andere Weise dumpen?
Der TensorBoard Graph Visualizer sieht nur die expliziten Gerätezuweisungen, die Sie in Ihrem Programm vorgenommen haben (d. h. diejenigen, die mit with tf.Device("..."): Blöcken erstellt wurden).
Der Grund dafür ist, dass die Knoten in einem TensorFlow-Graph Geräten in mehreren Stufen zugewiesen sind. Die erste Stufe im Client (z. B. Ihr Python-Programm) ermöglicht Ihnen die explizite und optionale Zuweisung von Geräten zu jedem Knoten, und es ist die Ausgabe dieser Stufe, die in die TensorBoard-Protokolle geschrieben wird. Eine spätere Platzierungsphase wird im TensorFlow-Back-End ausgeführt und weist jedem Knoten ein Gerät zu.
Ich vermute, dass Sie die Ergebnisse der späteren Placement-Phase analysieren möchten. Momentan gibt es keine Unterstützung dafür in TensorBoard, aber Sie können einige Informationen extrahieren, indem Sie tf.Session wie folgt erstellen:
... und dann werden die Geräteplatzierungsentscheidungen in stderr protokolliert.
Tags und Links tensorflow tensorboard