开始使用免费开始使用

双输出模型架构

在本练习中,您将构建一个多输出神经网络架构,用于同时预测字母表和字符。

回顾通用结构:在 .__init__() 方法中,您定义后续前向传播会用到的各层。在 forward() 方法中,先将输入图像依次通过若干层得到其嵌入向量(embedding),再将该嵌入分别送入两个独立的分类器层,每个输出对应一个分类器。

torch.nn 已按常用别名导入,现在让我们来构建模型吧!

本练习是课程的一部分

PyTorch 深度学习进阶

查看课程

交互式实操练习

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

class Net(nn.Module):
    def __init__(self):
        super().__init__()
        self.image_layer = nn.Sequential(
            nn.Conv2d(1, 16, kernel_size=3, padding=1),
            nn.MaxPool2d(kernel_size=2),
            nn.ELU(),
            nn.Flatten(),
            nn.Linear(16*32*32, 128)
        )
        # Define the two classifier layers
        ____ = ____
        ____ = ____
编辑并运行代码