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

Update wandb_utils.py (#4953) 

`is_valset_wandb_artifact` and `is_trainset_wandb_artifact` were referenced before assignment causing wandb to be unusable.
This commit is contained in:
Diego Montes 2021-09-28 15:22:31 -04:00 committed by GitHub
parent 38c779b099
commit 0c87478713
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -45,7 +45,8 @@ def check_wandb_config_file(data_config_file):
def check_wandb_dataset(data_file): def check_wandb_dataset(data_file):
is_wandb_artifact = False is_trainset_wandb_artifact = False
is_valset_wandb_artifact = False
if check_file(data_file) and data_file.endswith('.yaml'): if check_file(data_file) and data_file.endswith('.yaml'):
with open(data_file, errors='ignore') as f: with open(data_file, errors='ignore') as f:
data_dict = yaml.safe_load(f) data_dict = yaml.safe_load(f)