torchslim中在cifar10上的示例代码输出为64维，而不是10维

[wanghan0501]

问题如题，想请教下为什么最后一层使用conv而不是linear，改成conv会导致最终输出维度不定

self.linear=nn.Conv2d(self.base_channel*4*block.expansion,self.base_channel*4*block.expansion,1,1,0)
# self.linear = nn.Linear(self.base_channel*4*block.expansion, num_classes)

[gdh1995]

有道理啊。感觉这行是测试用的代码，提交时没改回去。可以手动改，或者直接用标准的 resnet，比如 torchvision.models.resnet18

[wanghan0501]

感谢，我尝试下。测试trt的时候被64维度弄懵逼了，还在想为什么会这样...

[wanghan0501]

还有一个问题，目前这个库不支持pytorch1.12版本，升级到1.12运行qat会报错