SROIE trained Donut model has repeating tokens in the prediction

philschmid / document-ai-transformers

MIT License

329 stars 50 forks source link

SROIE trained Donut model has repeating tokens in the prediction #8

Closed weiw05 closed 1 year ago

weiw05 commented 1 year ago

Hi, I'm following the code in your blog about custom training a Donut model (https://www.philschmid.de/fine-tuning-donut), but having issues. The training completes, but the prediction json has key "text_sequence" instead of the custom tokens, and has a bunch garbage as value. Any idea what I did wrong?

weiw05 commented 1 year ago

Looks like package version issues. Closing.