KRNN Code: https://github.com/microsoft/FOST/blob/main/fostool/model/krnn.py Introductions about the settings/configs. Torch_geometric is used in the original model in FOST, but we didn't use it. make use your CUDA version matches the torch version to allow the usage of GPU, we use CUDA==10.2 and torch.version==1.12.1