参考资料 ultralytics/utils/dist.py
备注
该文件可从https://github.com/ultralytics/ultralytics/blob/main/ ultralytics/utils/dist .py。如果您发现问题,请通过提交 Pull Request🛠️ 帮助修复。谢谢🙏!
ultralytics.utils.dist.find_free_network_port()
查找 localhost 上的空闲端口。
在单节点训练中,当我们不想连接到真正的主节点,但又必须设置
MASTER_PORT
环境变量。
源代码 ultralytics/utils/dist.py
ultralytics.utils.dist.generate_ddp_file(trainer)
生成 DDP 文件并返回其文件名。
源代码 ultralytics/utils/dist.py
ultralytics.utils.dist.generate_ddp_command(world_size, trainer)
生成并返回分布式训练指令。