search

megvii-research / TLC

Test-time Local Converter
Other
225 stars 10 forks source link

size mismatch #24

Open fengchenTM opened 8 months ago

fengchenTM commented 8 months ago

我输入一张图片大小为（2160，3672）， 经过class AvgPool2d的输出维度是（1， 1，1945, 3457），训练时用的AdaptiveAvgPool2d(1) 再经过flatten，fc就不能正常匹配维度了 请问该怎么修改 image

fengchenTM commented 8 months ago

TLC是不是只能用在用Conv来实现的全连接层上，不然局部平均池化的输出为（C，H，W）与全局平均池化的输出（C，1，1）对不上

achusky commented 8 months ago

是的，需要使用Conv实现的全连接层来处理(C,H,W)的输出。你可以将训练好的nn.Linear中的weight经过变换之后载入到重新实现的nn.Conv2d当中，并取消torch.flatten操作。

fengchenTM commented 8 months ago

哇，谢谢你如此快速的回复