如何修改pytorch网络的层数?

时间:2021-06-21 11:05:13

标签: python pytorch conv-neural-network layer

我正在使用这个模型:https://github.com/pytorch/examples/blob/0d3fe14a1c5a00795e3671ea3473caef6f0da72d/mnist/main.py 我想修改网络层的过滤器数量。我是pytorch的初学者。我怎样才能做到这一点? 以卷积层1为例:

self.conv1 = nn.Conv2d(1, 20, 5, 1)

如何修改过滤器的数量?

0 个答案:

没有答案
相关问题