bubbliiiing/unet-pytorch

导出onnx出错

jiejiangd opened this issue · 0 comments

在使用vgg的时候导出没错,但是使用resnet的时候导出报错
size mismatch for up_concat4.conv1.weight: copying a param with shape torch.Size([512, 3072, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 1024, 3, 3]).
size mismatch for up_concat3.conv1.weight: copying a param with shape torch.Size([256, 1024, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 768, 3, 3]).
size mismatch for up_concat2.conv1.weight: copying a param with shape torch.Size([128, 512, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 384, 3, 3]).