1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Bài tập

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.

Hướng dẫn

100 XP
  • 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_model từ lớp ManufacturingCNN().
  • Tải trọng số từ ModelCNN.pth vào loaded_model bằng cách truyền trọng số vào .load_state_dict().