मॉडल को सेव और लोड करें
एक मैन्युफैक्चरिंग कंपनी अपने प्रोजेक्ट्स को इमेज के आधार पर क्लासिफाई करना चाहती है और उसके अनुसार शिपिंग पैकेजिंग तय करना चाहती है. आपने PyTorch में एक हाई-एक्युरेसी मॉडल ट्रेन कर लिया है. अब आप मॉडल और उसके प्री-ट्रेंड वेट्स को भविष्य में उपयोग और अपनी टीम के साथ साझा करने के लिए सेव करने की योजना बनाते हैं, ताकि वे इसे बिना किसी दिक्कत के लोड कर सकें.
torch और torch.nn को nn नाम से इम्पोर्ट किया जा चुका है. प्री-ट्रेंड मॉडल ऑब्जेक्ट आपके वर्कस्पेस में model के रूप में उपलब्ध है, और इसकी आर्किटेक्चर ManufacturingCNN के रूप में उपलब्ध है.
यह अभ्यास पाठ्यक्रम का हिस्सा है
PyTorch के साथ इमेज के लिए डीप लर्निंग
अभ्यास निर्देश
- प्री-ट्रेंड मॉडल को
ModelCNN.pthनाम से सेव करें, ध्यान रखें कि सिर्फ आर्किटेक्चर नहीं, वेट्स सेव हों. ManufacturingCNN()क्लास सेloaded_modelनाम का एक मॉडल इंस्टैंस बनाएँ.ModelCNN.pthके वेट्स को.load_state_dict()में वेट्स पास करकेloaded_modelमें लोड करें.
इंटरैक्टिव व्यावहारिक अभ्यास
इस अभ्यास को इस नमूना कोड को पूरा करके आज़माएँ।
# Save the model
torch.____(model.____, ____)
# Create a new model
loaded_model = ____
# Load the saved model
loaded_model.____(torch.____('ModelCNN.pth'))
print(loaded_model)