search

SY-Xuan / DSCL

AAAI-24 Decoupled Contrastive Learning for Long-Tailed Recognition
20 stars 0 forks source link

distributed training #3

Open 1084991943 opened 1 week ago

1084991943 commented 1 week ago

Hi, I have only one gpu and can't do distributed training, is there a solution for this.

SY-Xuan commented 1 week ago

It is ok for single gpu training. You need to reduce the batch size to save the GPU memory. However, it may decrease the performance.