Diego Montes
commited on
Commit
•
0c87478
1
Parent(s):
38c779b
Update wandb_utils.py (#4953)
Browse files`is_valset_wandb_artifact` and `is_trainset_wandb_artifact` were referenced before assignment causing wandb to be unusable.
utils/loggers/wandb/wandb_utils.py
CHANGED
@@ -45,7 +45,8 @@ def check_wandb_config_file(data_config_file):
|
|
45 |
|
46 |
|
47 |
def check_wandb_dataset(data_file):
|
48 |
-
|
|
|
49 |
if check_file(data_file) and data_file.endswith('.yaml'):
|
50 |
with open(data_file, errors='ignore') as f:
|
51 |
data_dict = yaml.safe_load(f)
|
|
|
45 |
|
46 |
|
47 |
def check_wandb_dataset(data_file):
|
48 |
+
is_trainset_wandb_artifact = False
|
49 |
+
is_valset_wandb_artifact = False
|
50 |
if check_file(data_file) and data_file.endswith('.yaml'):
|
51 |
with open(data_file, errors='ignore') as f:
|
52 |
data_dict = yaml.safe_load(f)
|