开始使用免费开始使用

添加新的卷积层

您的项目负责人为您提供了一个新的 CNN 模型。先查看该模型的结构,然后为它追加一个新的卷积层。

模型以 CNNModel 提供。包 torchtorch.nn(简称 nn)已导入。

本练习是课程的一部分

使用 PyTorch 进行图像深度学习

查看课程

练习说明

  • CNNModel 类实例化一个模型,并访问其中的卷积层。
  • 创建一个新的卷积层:其 in_channels 等于现有层的 out_channelsout_channels 设为 32,stridepadding 都设为 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)
编辑并运行代码