PaddleOCR2Pytorch icon indicating copy to clipboard operation
PaddleOCR2Pytorch copied to clipboard

attention head weight shapes in paddle and pytorch model are different

Open iazdan opened this issue 3 years ago • 1 comments

I was able to convert the PP-structure paddle train model (best_accuracy.pdparams, best_accuracy.pdopt, best_accuracy.stated) to pytorch model (model.pth). However I realized that the model output is not the same as the paddle model. I did some investigation and I seems the paddle and the backbone weights are the same however the shape of the attention head in the pytorch model is different than the paddle model. This is a bug in the code. Would you be able to fix it?

Thanks

iazdan avatar Jun 16 '22 15:06 iazdan

table model is not supported.

frotms avatar Jun 18 '22 05:06 frotms