Lưu và tải một mô hình
Một công ty sản xuất muốn phân loại các dự án dựa trên hình ảnh và xác định loại bao bì vận chuyển phù hợp. Sau khi đã huấn luyện được một mô hình có độ chính xác cao trong PyTorch, bạn dự định lưu mô hình và các trọng số đã huấn luyện để dùng sau và chia sẻ với nhóm, đảm bảo họ có thể tải lại một cách trơn tru.
torch và torch.nn dưới tên nn đã được import. Đối tượng mô hình đã được huấn luyện sẵn có trong không gian làm việc của bạn là model, và kiến trúc của nó là ManufacturingCNN.
Bài tập này là một phần của khóa học
Deep Learning cho Ảnh với PyTorch
Hướng dẫn bài tập
- Lưu mô hình đã huấn luyện sẵn dưới tên
ModelCNN.pth, nhớ lưu phần trọng số chứ không chỉ kiến trúc. - Tạo một thể hiện mô hình tên
loaded_modeltừ lớpManufacturingCNN(). - Tải trọng số từ
ModelCNN.pthvàoloaded_modelbằng cách truyền trọng số vào.load_state_dict().
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Save the model
torch.____(model.____, ____)
# Create a new model
loaded_model = ____
# Load the saved model
loaded_model.____(torch.____('ModelCNN.pth'))
print(loaded_model)