https://medium.com/@yassine.hamdaoui/text-classification-using-transformers-pytorch-implementation-5ff9f21bd106