Mô hình phân loại nhị phân

Là một người làm deep learning, một trong những nhiệm vụ chính của bạn là huấn luyện các mô hình phân loại ảnh. Bạn thường gặp các bài toán phân loại nhị phân, nơi cần phân biệt giữa hai lớp. Để chuẩn hóa quy trình và dễ tái sử dụng, bạn quyết định tạo một mẫu (template) cho mô hình CNN phân loại ảnh nhị phân, có thể áp dụng cho các dự án sau này.

Gói torch và torch.nn dưới tên nn đã được import. Tất cả ảnh đều có kích thước 64x64 pixel.