Unverified 提交 cb2ad9f6 authored 作者: or-toledano's avatar or-toledano 提交者: GitHub

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

* Fix missing data folder when running log_dataset * Use ROOT/'data' * PEP8 whitespace
上级 c21da596
...@@ -356,7 +356,7 @@ class WandbLogger(): ...@@ -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:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论