1. Learn
    2. /
  2. Courses
    3. /
  3. Deep Learning nâng cao với PyTorch
Connected

Exercise

Vòng lặp huấn luyện bộ phân loại ảnh

Đến lúc huấn luyện bộ phân loại ảnh rồi! Bạn sẽ dùng Net đã định nghĩa trước đó và huấn luyện nó để phân biệt bảy loại mây.

Để định nghĩa hàm mất mát và bộ tối ưu, bạn sẽ dùng các hàm từ torch.nn và torch.optim, đã được nhập sẵn lần lượt là nn và optim. Bạn không cần thay đổi gì trong chính vòng lặp huấn luyện: nó giống hệt các vòng lặp bạn đã viết trước đây, chỉ thêm một chút logic để in loss trong quá trình huấn luyện.

Instructions

100 XP
  • Khởi tạo mô hình bằng lớp Net của bạn với num_classes đặt là 7 và gán cho net.
  • Định nghĩa hàm mất mát là cross-entropy loss và gán cho criterion.
  • Định nghĩa bộ tối ưu là Adam, truyền vào các tham số của mô hình và tốc độ học 0.001, rồi gán cho optimizer.
  • Bắt đầu vòng lặp for huấn luyện bằng cách lặp qua images và labels của dataloader_train.