टेन्सफोर्लो: model_with_buckets मॉडल में freeze_graph.py के लिए "output_node_names" क्या हैं?

Question

मैंने के साथ Tensorflow Tutorial में उदाहरण के समान tf.nn.seq2seq.model_with_buckets प्रशिक्षित किया।

अब मैं freeze_graph.py का उपयोग करके ग्राफ को फ्रीज करना चाहता हूं। मैं अपने मॉडल में "output_node_names" कैसे ढूंढ सकता हूं?

Answer 1

आप वैकल्पिक name="myname" तर्क को पारित करके अपने मॉडल में नोड्स के नामों का चयन कर सकते हैं जो कि किसी नोड का निर्माण करने वाले किसी भी Tensorflow ऑपरेटर को तर्क देता है। यदि आप उन्हें निर्दिष्ट नहीं करते हैं तो Tensorflow स्वचालित रूप से ग्राफ नोड्स के लिए नाम चुनेंगे, लेकिन यदि आप उन नोड्स को freeze_graph.py जैसे टूल पर पहचानना चाहते हैं, तो अपने नामों को चुनना सबसे अच्छा है। वे नाम हैं जिन्हें आप output_node_names पर पास करते हैं।

Answer 2

आप की तरह कुछ के साथ अपने मॉडल में नोड नाम के सभी प्राप्त कर सकते हैं:

node_names = [node.name for node in tf.get_default_graph.as_graph_def().node] 

या ग्राफ बहाल करने के साथ:

saver = tf.train.import_meta_graph(/path/to/meta/graph) 
sess = tf.Session() 
saver.resore(sess, /path/to/checkpoints) 
graph = sess.graph 
print([node.name for node in graph.as_graph_def().node]) 

आप इन फिल्टर करने के लिए केवल प्राप्त करने के लिए आवश्यकता हो सकती है आपके आउटपुट नोड्स, या केवल नोड्स जो आप चाहते हैं, लेकिन इससे कम से कम आपको उस ग्राफ के नाम प्राप्त करने में सहायता मिल सकती है जिसे आपने पहले ही प्रशिक्षित किया है और प्रत्येक नोड के लिए name='some_name' के साथ परिभाषित करने का जोखिम नहीं उठा सकता है।

आदर्श रूप से, आप प्रत्येक ऑपरेशन या टेंसर के लिए name पैरामीटर को परिभाषित करना चाहते हैं जिसे आप बाद में एक्सेस करना चाहते हैं।