import torch
from module.layer import simple_decoder,segmentation_head
#错误写法,未实例化直接调用
class model(nn.Module):
def __init__(self):
super(model, self).__init__()
def forward(self,x):
x=segmentation_head(x)
return x
# 正确写法,先实例化,再调用
class model(nn.Module):
def __init__(self):
super(model, self).__init__()
self.segmentation_head=segmentation_head()
def forward(self,x):
x=self.segmentation_head(x)
return x
Pytorch报错TypeError : __init__() takes 1 positional argument but 2 were given 原因及解决方法
最新推荐文章于 2023-07-04 18:03:03 发布