https://www.pylessons.com/handwriting-recognition-pytorch