构建卷积网络

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

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

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

torch 和 torch.nn as nn 已经为您导入好了，开始动手吧！

定义模型中的 feature_extractor：再添加一个具有 64 个输出特征图的卷积层、ELU 激活，以及窗口大小为 2 的最大池化层；最后将输出展平。