最近学习目标检测算法搭建网络的时候,检查程序遇到错误 ,如 TypeError: torch.FloatTensor is not a Module subclass 找了很长时间 总算解决了 。
错误如下:
class A(nn.Module): def __init__(self): super(A, self).__init__() self.module = nn.Sequential ( nn.Conv2d(), nn.BatchNorm2d(), nn.LeakyReLU() ) 改正如下 : class A(nn.Module): def __init__(self): super(A, self).__init__() self.module = nn.Sequential( nn.Conv2d(), nn.BatchNorm2d(), nn.LeakyReLU() )
总的来说是括号的原因,应该好像是是python的原因吧。