zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix `ROOT / data` when running W&B `log_dataset()` (#6606) 

* Fix missing data folder when running log_dataset

* Use ROOT/'data'

* PEP8 whitespace
This commit is contained in:
or-toledano 2022-02-10 17:17:47 +02:00 committed by GitHub
parent c21da596f3
commit cb2ad9f685
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/loggers/wandb/wandb_utils.py
View File

@ -356,7 +356,7 @@ class WandbLogger():
# create a _wandb.yaml file with artifacts links if both train and test set are logged # create a _wandb.yaml file with artifacts links if both train and test set are logged
if not log_val_only: if not log_val_only:
path = (path.stem if overwrite_config else path.stem + '_wandb') + '.yaml' # updated data.yaml path path = (path.stem if overwrite_config else path.stem + '_wandb') + '.yaml' # updated data.yaml path
path = Path('data') / path path = ROOT / 'data' / path
data.pop('download', None) data.pop('download', None)
data.pop('path', None) data.pop('path', None)
with open(path, 'w') as f: with open(path, 'w') as f: