这里是指对象不可以被调用,在设计网络架构尤其是自己写新的模块时容易出错,这里在进行定义时一定不要忘记声明nn.module,避免出现此类错误。(网页中没找到相似问题,因此记录一下)
源代码:
class block:
def __init__(self,channels):
super(block, self).__init__()......
修改后:
class block(nn.Module):
def __init__(self,channels):
super(block, self).__init__()......