BaşlayınÜcretsiz Başlayın

Bir modeli kaydet ve yükle

Bir üretim şirketi, projelerini görsellere göre sınıflandırmak ve uygun sevkiyat paketlemesini belirlemek istiyor. PyTorch ile yüksek doğrulukta bir model eğittikten sonra, şimdi modeli ve önceden eğitilmiş ağırlıklarını gelecekte kullanmak ve ekibinle paylaşmak için kaydetmeyi planlıyorsun; böylece onlar da modeli sorunsuzca yükleyebilsin.

torch ve torch.nn nn olarak içe aktarılmış durumda. Önceden eğitilmiş model nesnesi çalışma alanında model olarak, mimarisi ise ManufacturingCNN olarak mevcut.

Bu egzersiz

PyTorch ile Görüntüler için Deep Learning

kursunun bir parçasıdır
Kursu Görüntüle

Egzersiz talimatları

  • Önceden eğitilmiş modeli, yalnızca mimariyi değil ağırlıkları da kaydetmeyi unutmadan ModelCNN.pth olarak kaydet.
  • ManufacturingCNN() sınıfından loaded_model adlı bir model örneği oluştur.
  • .load_state_dict() metoduna ağırlıkları geçirerek ModelCNN.pth içindeki ağırlıkları loaded_modele yükle.

Uygulamalı interaktif egzersiz

Bu örnek kodu tamamlayarak bu egzersizi bitirin.

# Save the model
torch.____(model.____, ____)

# Create a new model
loaded_model = ____

# Load the saved model
loaded_model.____(torch.____('ModelCNN.pth'))
print(loaded_model)
Kodu Düzenle ve Çalıştır