Menyimpan dan memuat model

Sebuah perusahaan manufaktur ingin mengklasifikasikan proyek berdasarkan gambar dan menentukan kemasan pengiriman yang tepat. Setelah melatih model yang sangat akurat di PyTorch, Anda sekarang berencana untuk menyimpan model dan bobot pra-latihan untuk penggunaan di masa mendatang serta membagikannya kepada tim Anda, agar mereka dapat memuatnya tanpa hambatan.

torch dan torch.nn sebagai nn telah diimpor. Objek model pra-latihan tersedia di ruang kerja Anda sebagai model, dan arsitekturnya sebagai ManufacturingCNN.

Latihan ini adalah bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Petunjuk latihan

Simpan model pra-latihan sebagai ModelCNN.pth dengan mengingat untuk menyimpan bobotnya, bukan hanya arsitekturnya.
Buat instance model bernama loaded_model dari kelas ManufacturingCNN().
Muat bobot ModelCNN.pth ke loaded_model dengan meneruskan bobot tersebut ke .load_state_dict().

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

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

# Create a new model
loaded_model = ____

# Load the saved model
loaded_model.____(torch.____('ModelCNN.pth'))
print(loaded_model)

Edit dan Jalankan Kode