0

0

GGitHubadded KRNN and Sandwich models and their example results based on Alpha360 (#1414 )

efffb281创建于 2023年5月26日历史提交

Sandwich

Code: https://github.com/microsoft/FOST/blob/main/fostool/model/sandwich.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