search

BioinfoMachineLearning / DeepInteract

A geometric deep learning pipeline for predicting protein interface contacts. (ICLR 2022)
https://zenodo.org/record/6671582
GNU General Public License v3.0
62 stars 11 forks source link

the memory in each GPU seems to be different #20

Closed onlyonewater closed 1 year ago

onlyonewater commented 1 year ago

hi, authors, great work, when I used your code to train the model in the DDP form, the memory of each GPU was different, so how do I fix the code to let each GPU have the same memory when using DDP to train the model?