OpenGVLab/EfficientQAT

How to do e2e_qp with multi-GPU?

mxjmtxrm opened this issue · 1 comments

mxjmtxrm commented

As title. I set CUDA_VISIBLE_DEVICES=0,1,2,3 before python main_e2e_qp.py --xxxxx. I found that only GPU0 was used during training. Is there any parameters I missed?

ChenMnZ commented

Current code only support for the single-GPU training.