HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Import YOLOv5 dataloader (#94)

Browse source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Ayush Chaurasia 2022-12-26 06:00:01 +05:30 • committed by GitHub
parent ae05d44877
commit 16e3c08883
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
12 changed files with 1761 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ultralytics/yolo/utils/dist.py
View file

@ -47,7 +47,7 @@ def generate_ddp_command(world_size, trainer):
if using_cli:
file_name = generate_ddp_file(trainer)
return [
sys.executable, "-m", "torch.distributed.launch", "--nproc_per_node", f"{world_size}", "--master_port",
sys.executable, "-m", "torch.distributed.run", "--nproc_per_node", f"{world_size}", "--master_port",
f"{find_free_network_port()}", file_name] + sys.argv[1:]
@ -55,7 +55,7 @@ def ddp_cleanup(command, trainer):
# delete temp file if created
# TODO: this is a temp solution in case the file is deleted before DDP launching
time.sleep(5)
tempfile_suffix = str(id(trainer)) + ".py"
tempfile_suffix = f"{id(trainer)}.py"
if tempfile_suffix in "".join(command):
for chunk in command:
if tempfile_suffix in chunk: