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

原程序如下:

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)

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

又如

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))

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值