我正在使用“预训练图神经网络的策略”github 代码：https ://github.com/snap-stanford/pretrain-gnns 。训练后获得的“model.pth”文件与 DGL-lifesci 分子嵌入代码（ https://github.com/awslabs/dgl-lifesci/tree/master/examples/molecule_embeddings ）不兼容，作为文件内容不同。
任何人都可以帮助我提供与 DGL-LifeSci 中的 'gin-supervised-contextpred' 、 'gin-supervised-edgepred' 、 'gin-supervised-infomax' 路径文件兼容的训练代码吗？