Inference for token classification.

[Akhilesh64]

Hi @uakarsh, I trained docformer on a custom dataset and got a checkpoint file. I have no idea how to perform inference on test images. Can you show me how to do inference on images or if possible share a code snippet for the same.

[riteshKumarUMass]

I found this Kaggle notebook to be really helpful. You may refer to the last section of notebook for inferencing.

Did you pre-train the model ? Several of us found the model to be overfitting on training data without prior pre-training. Could you share you observation regarding the same?