yuyangw / MolCLR

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

multi GPU trainging problem

MrsW6 opened this issue · comments

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

the source sentence is
image

could u please give me some suggestions?