yolox训练模型时,只能用一个GPU
一共有三个地方可以设置
train.py
parser.add_argument(
"-d", "--devices", default=4, type=int, help="device for training"
launch.py
def launch(
main_func,
num_gpus_per_machine,
num_machines=4,
machine_rank=0,
backend="nccl",
dist_url=None,
args=(),
timeout=DEFAULT_TIMEOUT,
开始训练指令
python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 0 -d 4 -b 16 --fp16 -o -c weights/yolox_s.pth
各种奇怪报错指令
没有办法了
能够实现多GPU训练(4GPU)
你好,我是有问必答小助手,非常抱歉,本次您提出的有问必答问题,技术专家团超时未为您做出解答
本次提问扣除的有问必答次数,已经为您补发到账户,我们后续会持续优化,扩大我们的服务范围,为您带来更好地服务。