Lưu và nạp state_dict
Hiểu state_dict là phần trọng tâm để lưu model artifact và nạp lại ở nơi khác.
Tiến độ của tôi ở bài này
Điểm được lưu vào tài khoản sau khi chấm bài.
Đang tải điểm của bạn…
Kiến thức và chủ đề
Kiến thức tiên quyết: nn.Linear, forward, model parameters.
Nội dung đề bài
Mục tiêu kiến thức
Hiểu state_dict là phần trọng tâm để lưu model artifact và nạp lại ở nơi khác.
Mô tả bài toán
Đây là bài số 393 trong tầng PyTorch deep learning thực hành. Bài dùng thư viện torch.save + load_state_dict + io.BytesIO trên dữ liệu nhỏ, cố định và chạy CPU; mục tiêu là hiểu workflow chứ không bắt học viên tự viết lại thư viện.
Yêu cầu
Tạo Linear model, lưu state_dict vào buffer, tạo model mới, load state_dict và so output trước/sau.
def save_load_state_dict(X):Input
X là batch số.
Output
Dictionary state_keys, max_abs_diff, output_before, output_after.
Ràng buộc
max_abs_diff sau round trip phải bằng 0; chỉ lưu state_dict, không lưu toàn object model.
Ví dụ
Giải thích
Output của model gốc và model đã load phải giống hệt trong sai số số học.
Gợi ý và lời giải chỉ mở sau khi bạn bấm Nộp bài. Giáo viên và quản trị viên mở được ngay.
Góp ý & báo lỗi bài tập
Đề bài chưa rõ, test có vấn đề hay bạn có ý tưởng giúp bài tốt hơn? Gửi cho đội ngũ AI Empire nhé — mỗi góp ý đều được đọc.