Tạo một khối tuần tự
Bạn quyết định thiết kế lại mẫu mô hình CNN nhị phân của mình bằng cách tạo một khối các tầng tích chập. Cách này giúp bạn xếp chồng nhiều tầng theo thứ tự. Với mô hình cải tiến này, bạn sẽ dễ dàng thiết kế nhiều kiến trúc CNN khác nhau.
torch và torch.nn dưới tên nn đã được nhập.
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
- Trong phương thức
__init__(), định nghĩa một khối các tầng tích chập và gán choself.conv_block. - Trong lượt truyền
forward(), đưa đầu vào đi qua khối tích chập bạn đã định nghĩa.
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.
class BinaryImageClassification(nn.Module):
def __init__(self):
super(BinaryImageClassification, self).__init__()
# Create a convolutional block
self.conv_block = ____(
nn.Conv2d(3, 16, kernel_size=3, stride=1, padding=1),
nn.ReLU(),
nn.Conv2d(16, 32, kernel_size=3, stride=1, padding=1),
nn.ReLU(),
)
def forward(self, x):
# Pass inputs through the convolutional block
x = ____
return x