This code is taken from https://github.com/rdspring1/PyTorch_GBW_LM.
The corresponding LICENSE has been added.
All credit goes to @rdspring1.