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.
Bài tập này là một phần của khóa học
Deep Learning cho Ảnh với PyTorch
Hướng dẫn bài tập
- Khởi tạo một mô hình từ lớp
CNNModelvà 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_channelsbằngout_channelscủa lớp hiện có, đặtout_channelslà 32,stridevàpaddingđều bằng 1, vàkernel_sizelà 3; gán vàoconv2. - Nối thêm lớp mới vào mô hình, đặt tên là
"conv2".
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Create a model
model = ____
print("Original model: ", model)
# Create a new convolutional layer
conv2 = ____
# Append the new layer to the model
model.____(____)
print("Extended model: ", model)