search

gaopengcuhk / Stable-Pix2Seq

A full-fledged version of Pix2Seq
Apache License 2.0
235 stars 20 forks source link

trained model #12

Open opentld opened 2 years ago

opentld commented 2 years ago

Is there any trained model for testing? Thank you！

opentld commented 2 years ago

I used the unofficial-Pix2Seq trained model , and got an error: RuntimeError: Error(s) in loading state_dict for DETR: size mismatch for transformer.embedding.position_embeddings.weight: copying a param with shape torch.Size([500, 256]) from checkpoint, the shape in current model is torch.Size([501, 256]).