Multi gpu memory imbalance

Megvii-BaseDetection / YOLOX

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

Apache License 2.0

9.34k stars 2.19k forks source link

Multi gpu memory imbalance #1329

Open emjay73 opened 2 years ago

emjay73 commented 2 years ago

When training, I'm suffering from multi-GPU memory imbalance Any idea?

FateScript commented 2 years ago

Is this long time or short time watching ?