添加新的卷积层
您的项目负责人为您提供了一个新的 CNN 模型。先查看该模型的结构,然后为它追加一个新的卷积层。
模型以 CNNModel 提供。包 torch 和 torch.nn(简称 nn)已导入。
本练习是课程的一部分
使用 PyTorch 进行图像深度学习
练习说明
- 由
CNNModel类实例化一个模型,并访问其中的卷积层。 - 创建一个新的卷积层:其
in_channels等于现有层的out_channels,out_channels设为 32,stride与padding都设为 1,kernel_size设为 3;将其赋值给conv2。 - 将该新层追加到模型中,并将其命名为
"conv2"。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# 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)