Imported from https://github.com/aksub99/MolDQN-pytorch