salesforce / awd-lstm-lm

LSTM and QRNN Language Model Toolkit for PyTorch
BSD 3-Clause "New" or "Revised" License
1.96k stars 488 forks source link

Update codebase to work with PyTorch 0.2 #3

Closed Smerity closed 7 years ago

Smerity commented 7 years ago

The original codebase was written to be run on PyTorch 0.1.12_2. Updating the codebase to work on PyTorch 0.2 requires a number of steps, including modifying WeightDrop and others.

Best would be to provide two branches - one with the current PyTorch 0.1.12_2 codebase (allowing for exact result replication) and a second branch that is updated to allow for PyTorch 0.2.