Current code is developed on PyTorch 0.4, not sure if it works on other versions.
A subset of data (20k docs) is provided here for you to test the code. Unzip and place it to data/.
If you need to train on the whole kp20k dataset, download the json data and run preprocess.py
first. No trained model will be released in the near future.