PyTorch ऑप्टिमाइज़र का उपयोग

पहले, आपने नेटवर्क के वेट को मैन्युअली अपडेट किया था, ताकि आपको पर्दे के पीछे ट्रेनिंग कैसे चलती है यह समझ आए। लेकिन कई लेयर्स वाले डीप नेटवर्क के लिए यह तरीका स्केलेबल नहीं है.

अच्छी बात यह है कि PyTorch का SGD ऑप्टिमाइज़र यह प्रक्रिया कुछ ही लाइनों के कोड में कुशलतापूर्वक ऑटोमेट कर देता है। अब, आप PyTorch ऑप्टिमाइज़र से वेट्स अपडेट करके ट्रेनिंग लूप को पूरा करेंगे.

एक न्यूरल नेटवर्क बनाया गया है और model वैरिएबल के रूप में उपलब्ध है। इसी मॉडल से फ़ॉरवर्ड पास चलाकर प्रेडिक्शंस का टेन्सर pred बनाया गया है। वन-हॉट एनकोडेड टेन्सर का नाम target है और क्रॉस-एंट्रॉपी लॉस फंक्शन criterion में स्टोर है.

torch.optim को optim नाम से, और torch.nn को nn नाम से पहले ही लोड कर दिया गया है.