问题:Pytorch报错TypeError : __init__() takes 1 positional argument but 2 were given
解决方法: 在网上搜了下,都是说自己的模型定义错误,我看了下,发现也没有错误,就很懵!
然后看看之前的代码发现我没有实例化!!!
贴代码:
class Cnn(nn.Module):
def __init__(self):
super(Cnn, self).__init__()
self.Conv = nn.Sequential(
Conv2dSame(4, 64, 5),
nn.ReLU(),
Conv2dSame(64, 128, 4),
nn.ReLU(),
Conv2dSame(128, 256, 4),
nn.ReLU(),
Conv2dSame(256, 64, 1),
nn.ReLU(),
nn.MaxPool2d(2, 2),
nn.Flatten(),
nn.Linear(1024, 512),
nn.ReLU()
)
def forward(self, x):
x = self.Conv(x)
return x
这个模块没有毛病。
可总是报错:__init__() takes 1 positional argument but 2 were given
在迭代训练之前加入 实例化这个网络文章来源:https://www.toymoban.com/news/detail-649404.html
net = Cnn() # <------------写上这个就行了
#下面就是训练的部分了
for index in range(EPOCH):
for i, (x, y) in enumerate(train_dataloader):
。。。
要是别的方法不行,可以试一试我的方法! 要说成功解决了,点个赞不过分吧,哈哈哈文章来源地址https://www.toymoban.com/news/detail-649404.html
到了这里,关于Pytorch报错TypeError : __init__() takes 1 positional argument but 2 were given 原因及解决方法的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!