1. Learn
    2. /
  2. 课程
    3. /
  3. PyTorch 深度学习进阶
Connected

道练习

双输入模型

数据已就绪，现在来构建双输入模型的架构！为此，您将创建一个模型类，包含以下方法：

  • .__init__()：在这里通过分组层来定义子网络；您需要定义两个用于处理两个输入的层，以及一个对每个类别返回分类分数的分类器。

  • forward()：在这里将两个输入分别传入对应的预定义子网络，拼接它们的输出，再传给分类器。

torch.nn 已作为 nn 为您导入。开始吧！

说明 1 / 共 3 个

undefined XP
    1
    2
    3
  • 将图像、字母和分类器子网络定义为顺序模型，并分别赋给 self.image_layer、self.alphabet_layer 和 self.classifier。