报错信息:卷积的输入应该是tensor,而不是Nonetype。
输入的图片是正常的tensor,但是为什么会报错。
通过定位与检测,发现我写一个自定义常规卷积的时候,忘记return了。
class BasicBlock(nn.Module):
expansion=1
def __init__(self):
super(BasicBlock, self).__init__()
self.conv1=nn.Conv2d(3,3,kernel_size=3,stride=stride,padding=1,bias=False)
self.bn1=nn.BatchNorm2d(3)
self.relu=nn.ReLU()
def forward(self,x):
x=self.conv1(x)
x=self.bn1(x)
x=self.relu(x)
# forward函数就写到了这里,然后没有return所以报错
#return out
将return加上即可