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

ultralytics 8.0.162 Multi-GPU DDP fix (#4544)

Browse source 
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: andresinsitu <andres.rodriguez@ingenieriainsitu.com>
This commit is contained in:
Glenn Jocher 2023-08-24 13:13:49 +02:00 • committed by GitHub
parent 1db9afc2e5
commit 2bcee56e70
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 24 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/data/utils.py
View file

@ -343,7 +343,7 @@ def check_cls_dataset(dataset: str, split=''):
# Print to console
for k, v in {'train': train_set, 'val': val_set, 'test': test_set}.items():
prefix = f'{colorstr(k)} {v}...'
prefix = f'{colorstr(f"{k}:")} {v}...'
if v is None:
LOGGER.info(prefix)
else: