TypeError: 'Net' object is not callable python报错-蒲公英云

TypeError: 'Net' object is not callable python报错

水深无声 2022-01-31 04:35 399阅读 0赞

原程序如下：

import torch
import torch.nn as nn
class Net():
    def __init__(self,inc,k):
        super().__init__()
        self.net = nn.Sequential(
            nn.Conv2d(inc,inc,k,2,1)
        )
    def forward(self,x):
        return self.net(x)
class UpsampleLayer(torch.nn.Module):
    def __init__(self):
        super(UpsampleLayer, self).__init__()
        self.pixelShuffle = torch.nn.PixelShuffle(2)
    def forward(self, x):
        return self.pixelShuffle(x)
if __name__ == '__main__':
    x = torch.Tensor(1,4,100,100)
    net1 = Net(4,3)
    y = net1(x)
    up = UpsampleLayer()
    y1 = up(y)
    print(y1)

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5OTM4NjY2_size_16_color_FFFFFF_t_70

按照网上常见的说法是命名冲突问题，但是我只是一个如此简单的demo，又没有占用关键字，怎么可能是命名问题。于是在第26行把 y = net1(x) 改为 y = net1.forward(x) ，告诉这个网络用其内的什么函数，就好了。。。

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM5OTM4NjY2_size_16_color_FFFFFF_t_70 1

又如

import torch
import torch.nn as nn
class Net():
    def __init__(self,inc,k):
        super().__init__()
        self.net = nn.Sequential(
            nn.Conv2d(inc,inc,k,2,1)
        )
    def forward(self,x):
        return self.net(x)
class UpsampleLayer(torch.nn.Module):
    def __init__(self):
        super().__init__()
        self.pixelShuffle = torch.nn.PixelShuffle(2)
    def forward(self, x):
        return self.pixelShuffle(x)
class MainNet():
    def __init__(self):
        super().__init__()
        self.net1 = Net(4, 3)
        self.up = UpsampleLayer()
    def forward(self,x):
        y = self.net1.forward(x)
        return self.up(y)
if __name__ == '__main__':
    x = torch.Tensor(1,4,100,100)
    mnet = MainNet()
    print(mnet.forward(x))