2015-10-22 9 views
9

मैं टॉर्च/लुआ प्रोजेक्ट पर काम कर रहा हूं, जिसमें मैंने एक कृत्रिम तंत्रिका नेटवर्क मॉडल लागू किया था। सब कुछ काम करता है, लेकिन अब मैं अपने कोड को निम्न तरीके से संशोधित करना चाहता हूं। चूंकि मेरा इनपुट डेटासेट बहुत बड़ा है, इसलिए मैं इसे एन = 20 स्पैन में विभाजित करना चाहता हूं।मशाल/लुआ, एक प्रशिक्षित तंत्रिका नेटवर्क मॉडल को फाइल करने के लिए कैसे बचाएं?

फिर मैं अपने तंत्रिका नेटवर्क को केवल प्रथम डेटासेट अवधि पर प्रशिक्षित करना चाहता हूं और फिर समानांतर में अन्य एन -1 = 1 9 स्पैन पर परीक्षण करना चाहता हूं।

इन सभी समांतर नौकरियों को चलाने के लिए, मुझे अपने तंत्रिका नेटवर्क मॉडल विवरण को फ़ाइल में सहेजने की आवश्यकता है, और फिर इसे हर 1 9 नौकरियों के लिए लोड करें।

क्या मशाल में फ़ाइल करने के लिए एक कृत्रिम तंत्रिका नेटवर्क मॉडल को "लिखने" के लिए कोई रास्ता है?

उत्तर

15
-- save the model 
torch.save(filename, model) 

-- load the model 
model = torch.load(filename) 
+0

धन्यवाद @ एसएमएचएक्स, मैंने सोचा जितना आसान! –

+0

क्या इसे सहेजने और इसे लुआ (बिना मशाल के) में लोड करने का कोई तरीका है? नेटवर्क को वापस मशाल में लोड करने से पहले – Adam12344

+0

@ Adam12344, आपको धारावाहिक मॉडल में उपयोग किए गए पैकेजों की आवश्यकता होती है, जैसे 'nn', अन्यथा लोडिंग विफल हो जाएगी। नतीजतन, यदि आप लोड नहीं कर सकते हैं तो यह निश्चित रूप से 'लुआ (बिना मशाल के) के साथ भी असफल हो जाएगा। मॉडल लोड करने से पहले 'nn' पैकेज। –

संबंधित मुद्दे