here is the dataset link: https://www.kaggle.com/datasets/robikscube/textocr-text-extraction-from-images-dataset I was struggling trying to adapt the code for the dataset above, the model would freeze after 1 full epoch