AttributeError: ‘NoneType‘ object has no attribute ‘dim BN层
大早上的一来就开始写bug。
def forward(self, x):
x = self.conv_stem(x)
x = self.bn1(x)
x = self.act1(x)
x = self.blocks(x)
x = self.global_pool(x)
x = self.conv_head(x)
x = self.act2(x)
x = x.view(x.size
原创
2020-09-20 10:12:10 ·
3342 阅读 ·
5 评论