开始使用免费开始使用

构建卷积网络

您所在的团队正在构建一套天气预报系统。系统的一部分是在各个地点安装摄像头,拍摄天空的照片。您的任务是训练一个模型,对照片中的不同云层类型进行分类,以帮助识别即将到来的天气锋面。

您决定构建一个卷积图像分类器。该模型包含两个部分:

  • 特征提取器:从输入图像中学习得到一个特征向量;
  • 分类器:根据学到的特征预测图像的类别。

torchtorch.nn as nn 已经为您导入好了,开始动手吧!

本练习是课程的一部分

PyTorch 深度学习进阶

查看课程

交互式实操练习

通过完成这段示例代码来试试这个练习。

class Net(nn.Module):
    def __init__(self, num_classes):
        super().__init__()
        # Define feature extractor
        self.feature_extractor = nn.Sequential(
            nn.Conv2d(3, 32, kernel_size=3, padding=1),
            nn.ELU(),
            nn.MaxPool2d(kernel_size=2),
            ____,
            ____,
            ____,
            ____,
        )
编辑并运行代码