Imported from https://github.com/malteos/pytorch-bert-document-classification