search

bubbliiiing / yolox-pytorch

这是一个yolox-pytorch的源码，可以用于训练自己的模型。
Apache License 2.0
880 stars 183 forks source link

训练加载慢正常吗？ #112

Closed YinZhang98 closed 2 years ago

YinZhang98 commented 2 years ago

CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 train.py 一直在:Load weights model_data/yolox_s.pth.