search

mkocabas / EpipolarPose

Self-Supervised Learning of 3D Human Pose using Multi-view Geometry (CVPR2019)
Other
594 stars 97 forks source link

multi gpu imbalance #22

Closed yuzhenbo closed 4 years ago

yuzhenbo commented 4 years ago

When i use four gpu to train this model, one the first gpu, Memory Usage is about 10761MB/12196M, while the other GPU all 3651M/12196M, which waste too much memory usage. Why do not you use GPU balance function (like DataParallelModel in Integral LOSS)? or can you account for the same situation?

yuzhenbo commented 4 years ago

may be i found some solution.

ericzw commented 4 years ago

@yuzhenbo hi, can you share me your solutions? Thanks