Salva grafico Tensorflow per la visualizzazione in Tensorboard senza operazioni di riepilogo

Question

Ho un grafico Tensorflow piuttosto complicato che vorrei visualizzare per scopi di ottimizzazione. C'è una funzione che posso chiamare che salverà semplicemente il grafico per la visualizzazione in Tensorboard senza dover annotare le variabili?

ho provato questo:

merged = tf.merge_all_summaries() 
writer = tf.train.SummaryWriter("/Users/Name/Desktop/tf_logs", session.graph_def) 

Ma nessuna uscita è stata prodotta. Questo sta usando la ruota 0.6.

questo sembra essere correlato: Graph visualisaton is not showing in tensorboard for seq2seq model

Answer 1

Per l'efficienza, i tf.train.SummaryWriter log in modo asincrono su disco. Per garantire che il grafico venga visualizzato nel registro, è necessario chiamare close() o flush() sul writer prima che il programma venga chiuso.

Answer 2

È inoltre possibile scaricare il grafico come un protobuf GraphDef e caricarlo direttamente in TensorBoard. Puoi farlo senza avviare una sessione o eseguire il modello.

## ... create graph ... 
>>> graph_def = tf.get_default_graph().as_graph_def() 
>>> graphpb_txt = str(a.graph.as_graph_def()) 
>>> with open('graphpb.txt', 'w') as f: f.write(graphpb_txt) 

In questo modo verrà generato un file simile a questo, in base alle specifiche del modello.

node { 
    name: "W" 
    op: "Const" 
    attr { 
    key: "dtype" 
    value { 
     type: DT_FLOAT 
    } 
    } 
... 
version 1 

In TensorBoard è quindi possibile utilizzare il pulsante "Carica" ​​per caricarlo dal disco.

Answer 3

Per tutta chiarezza, questo è come ho usato il metodo .flush() e risolto il problema:

Inizializzare lo scrittore con:

writer = tf.train.SummaryWriter("/home/rob/Dropbox/ConvNets/tf/log_tb", sess.graph_def) 

e utilizzare lo scrittore con:

writer.add_summary(summary_str, i) 
    writer.flush() 

Answer 4

Questo ha funzionato per me:

graph = tf.Graph() 
with graph.as_default(): 
    ... build graph (without annotations) ... 
writer = tf.summary.FileWriter(logdir='logdir', graph=graph) 
writer.flush() 

Il grafico viene caricato automaticamente all'avvio del tensore con "--logdir = logdir /". Nessun pulsante "upload" necessario.