Tensorflow 0.8 Importazione ed esportazione di problemi di tensioni di uscita

Question

Sto utilizzando Tensorflow 0.8 con Python 3. Sto cercando di addestrare la rete neurale e l'obiettivo è di esportare/importare automaticamente gli stati della rete ogni 50 iterazioni. Il problema è quando esporto il tensore di uscita alla prima iterazione, il nome del tensore di uscita è ['Neg:0', 'Slice:0'], ma quando esporto il tensore di uscita alla seconda iterazione, il nome del tensore di uscita viene modificato come ['import/Neg:0', 'import/Slice:0'] e l'importazione di questo tensore di uscita non funziona quindi:

ValueError: Specified colocation to an op that does not exist during import: import/Variable in import/Variable/read 

Mi chiedo se qualcuno ha idee su questo problema. Grazie!!!

Answer 1

Ecco come funziona tf.import_graph_def.

Se non si desidera il prefisso, è sufficiente impostare il parametro name sulla stringa vuota come mostrato nell'esempio seguente.

# import the model into the current graph 
with tf.Graph().as_default() as graph: 

    const_graph_def = tf.GraphDef() 
    with open(TRAINED_MODEL_FILENAME, 'rb') as saved_graph: 
     const_graph_def.ParseFromString(saved_graph.read()) 
     # replace current graph with the saved graph def (and content) 
     # name="" is important because otherwise (with name=None) 
     # the graph definitions will be prefixed with import. 
     # eg: the defined operation FC2/unscaled_logits:0 
     # will be import/FC2/unscaled_logits:0 
     tf.import_graph_def(const_graph_def, name="") 
    [...]