标签：forward

  • pytorch:我们可以直接在forward()函数中使用nn.Module层吗？ - python

    时间：2020-9-1

    通常，在构造函数中，我们声明要使用的所有层。在前向功能中，我们定义了如何从输入到输出运行模型。 我的问题是，如果直接在forward()函数中调用那些预定义/内置的 nn。这种Keras function API样式适用于 Pytorch 吗？如果没有，为什么？ 更新:以这种方式构造的TestModel 确实运行成功，没有发出警报。但是与传统方法相比，训练损 […]

    标签：, , ,