问题描述:
【功能模块】
model.train无法进行,没有到底是什么错误的提示
【操作步骤&问题现象】
1、
2、
【截图信息】
def construct(self, x): x = self.conv1(x) x = self.bn1(x) x = self.relu(x) x = self.maxpool(x) x = self.layer1(x) x = self.layer2(x) x = self.layer3(x) x = self.layer4(x)#(b,2048,25,12)输入到赤化层,编程(6,1),可以自己算一次 #youdianwen题,输出有点不太对啊 # 这里写怎么设置 # sx = int(x.shape[2] / self.part) # import math # sx=x.shape[2] //self.part#4 # # kx = x.shape[2] - sx * (self.part-1)#5 # # x = F.avg_pool2d(x, kernel_size=(kx, x.size(3)), stride=(sx, x.size(3))) # # avgpool=ops.AvgPool(ksize=(kx, x.shape[3]),strides=(sx, x.shape[3])) # #如果写死了,就不能去改变输入的尺寸 # pool=nn.AvgPool2d(kernel_size=(kx, x.shape[3]),s