TypeError: torch.FloatTensor is not a Module subclass

最近学习目标检测算法搭建网络的时候,检查程序遇到错误 ,如 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的原因吧。

你可能感兴趣的:(python)