Khối bộ phân loại
Nhiệm vụ tiếp theo của bạn là tạo một khối bộ phân loại để thay thế bộ phân loại VGG16 gốc. Bạn quyết định dùng một khối gồm hai lớp kết nối đầy đủ với một kích hoạt ReLU ở giữa.
vgg_model và input_dim mà bạn đã định nghĩa ở bài trước có sẵn trong không gian làm việc của bạn, và torch cùng torchvision.models đã đượ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
- Tạo biến
num_classesvới số lượng lớp, giả sử bạn chỉ phát hiện mèo và chó. - Tạo một khối tuần tự dùng
nn.Sequential. - Tạo một lớp tuyến tính với
in_featuresđặt làinput_dim. - Thêm số đặc trưng đầu ra cho lớp cuối cùng của bộ phân loại.
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 variable with the number of classes
____
# Create a sequential block
classifier = ____(
# Create a linear layer with input features
____(____, 512),
nn.ReLU(),
# Add the output dimension to the classifier
nn.Linear(512, ____),
)