标签: 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)
如何修改过滤器的数量?