Multi gpu stuck after the first epoch when training yolox

YuHengsss / YOLOV

This repo is an implementation of PyTorch version YOLOV Series

Apache License 2.0

309 stars 43 forks source link

Multi gpu stuck after the first epoch when training yolox #40

Closed kiter-zero closed 1 year ago

kiter-zero commented 1 year ago

I see the paper where 2/4 gpus can directly used to train the yolox model. However, I suffer from the stuck problem of multi gpu when training yolox model. Does anyone know how to solve this problem? Thank you!

kiter-zero commented 1 year ago

According to the install step provided in readme.md, the problem solve.