multi GPU trainging problem

yuyangw / MolCLR

Implementation of MolCLR: "Molecular Contrastive Learning of Representations via Graph Neural Networks" in PyG.

MIT License

233 stars 57 forks source link

Closed MrsW6 closed 1 year ago

MrsW6 commented 1 year ago

Hi, when I use your code to pretrain a GIN network using multi GPU,

the source sentence is

could u please give me some suggestions?