https://github.com/pytorch/examples/blob/master/word_language_model/main.py#L178
https://github.com/pytorch/examples/blob/master/word_language_model/main.py#L178