sreejith3534

https://github.com/nlpyang/pytorch-transformers/blob/master/examples/utils_squad_evaluate.py @yzgdjqwh ,I guess you are looking for this ?

> **Model I am using : LayoutLM** > > During the training, I am getting CUDA device side assert error while performing the forward pass. Attached below is the screenshot...

Hi @Allen-Shao , How did you tried out prediction providing the input image. Can you tell me steps involved?