The model was fine-tuned on the dataset available in https://huggingface.co/datasets/U20212419/digits, using the model ResNet-18 available in PyTorch, based on He et al. (2015).

He, K., Zhang, X., Ren, S., & Sun, J. (2015). Deep Residual Learning for Image Recognition. https://arxiv.org/abs/1512.03385