Requirements Latest tested combination: Python 3.8.1 + PyTorch 1.5.0 + DGL 0.8.1. How to run run drug_train.py