https://huggingface.co/learn/nlp-course/chapter1/3?fw=pt https://colab.research.google.com/github/huggingface/notebooks/blob/master/course/en/chapter1/section3.ipynb