pytorch resnet去掉池化层和全连接层