ocr_vit refer to https://github.com/chinhsuanwu/mobilevit-pytorch https://github.com/Deepayan137/Adapting-OCR/tree/master/src