Referencia para ultralytics/utils/dist.py
Nota
Este archivo está disponible en https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/utils/dist .py. Si detectas algún problema, por favor, ayuda a solucionarlo contribuyendo con una Pull Request 🛠️. ¡Gracias 🙏!
ultralytics.utils.dist.find_free_network_port()
Encuentra un puerto libre en localhost.
Es útil en el entrenamiento de un solo nodo cuando no queremos conectarnos a un nodo principal real pero tenemos que establecer el
MASTER_PORT
variable de entorno.
Código fuente en ultralytics/utils/dist.py
ultralytics.utils.dist.generate_ddp_file(trainer)
Genera un archivo DDP y devuelve su nombre.
Código fuente en ultralytics/utils/dist.py
ultralytics.utils.dist.generate_ddp_command(world_size, trainer)
Genera y devuelve el comando para el entrenamiento distribuido.