Pytorch报错TypeError : __init__() takes 1 positional argument but 2 were given 原因及解决方法

问题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
在迭代训练之前加入 实例化这个网络

net  = Cnn()    #   <------------写上这个就行了
#下面就是训练的部分了
for index in range(EPOCH):
      for i, (x, y) in enumerate(train_dataloader):
 	  。。。

要是别的方法不行,可以试一试我的方法! 要说成功解决了,点个赞不过分吧,哈哈哈

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
xiaoxingxing的头像xiaoxingxing管理团队
上一篇 2023年7月29日
下一篇 2023年7月29日

相关推荐