使用Pytorch的自定义NN体系结构

Question

我正在尝试使用Pytorch创建自定义的CNN架构。 我希望拥有与仅使用numpy进行架构时得到的控件相同的控件。

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

Answer 1

您必须在自定义类中实现backward()函数。

但是从您的问题尚不清楚

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