python - 使用 Pytorch 的自定义 NN 架构

Question

我正在尝试使用 Pytorch 制作自定义 CNN 架构。如果我只使用 numpy 制作架构，我希望拥有与我将获得的控制大致相同的控制。

我是 Pytorch 的新手，如果可能的话，我希望看到一些不使用 nn.module 类实现的 CNN 代码示例。

标签： pythonneural-networkconv-neural-networkpytorch

Answer 1

您必须在自定义类中实现backward()功能。

但是，从您的问题来看，尚不清楚是否

• 你只需要一系列新的 CNN 块（所以你最好使用 nn.module 和类似的东西 nn.Sequential(nn.Conv2d( ...) ）
• 你只需要梯度下降 https://github.com/jcjohnson/pytorch-examples#pytorch-autograd，所以你自己向后计算。