shunsunsun / 3dinfomax Goto Github PK
View Code? Open in Web Editor NEWThis project forked from hannesstark/3dinfomax
Making self-supervised learning work on molecules by using their 3D geometry to pre-train GNNs. Implemented in DGL and Pytorch Geometric.