Imported from https://github.com/Saner3/pytorch-transformers-comet