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

Bài tập

Thêm một lớp tích chập mới

Trưởng dự án đã cung cấp cho bạn một mô hình CNN mới. Hãy xem kiến trúc của mô hình và nối thêm một lớp tích chập mới vào đó.

Mô hình có sẵn dưới tên CNNModel. Các gói torch và torch.nn với bí danh nn đã được import.

Hướng dẫn

100 XP
  • Khởi tạo một mô hình từ lớp CNNModel và truy cập các lớp tích chập.
  • Tạo một lớp tích chập mới với in_channels bằng out_channels của lớp hiện có, đặt out_channels là 32, stride và padding đều bằng 1, và kernel_size là 3; gán vào conv2.
  • Nối thêm lớp mới vào mô hình, đặt tên là "conv2".