zoukankan      html  css  js  c++  java
  • YOLOX multiGPU training

    将yolox/core/trainer.py中的self.tblogger.add_scalar 语句放在if self.rank == 0: 中,结束后需要synchronize()
            if self.rank == 0:
                self.tblogger.add_scalar('lr', lr, self.epoch)
                self.tblogger.add_scalar('loss', loss, self.epoch)
            Add :synchronize()
    #4个 GPU 训练的命令
    #python -m torch.distributed.launch --nproc_per_node=4 tools/train.py -n exps/example/yolox_voc/yolox_voc_s.py -b 32

    人生,从没有一劳永逸 想要变强,只有不停奔跑
  • 相关阅读:
    JasperReport
    Linux
    Linux
    Linux
    Linux
    Groovy
    Linux
    VS
    Aliyun
    Linux
  • 原文地址:https://www.cnblogs.com/jimchen1218/p/15413564.html
Copyright © 2011-2022 走看看