W&B: Fix for 4360 (#4388)
* Improve docstrings and run names * default wandb login prompt with timeout * return key * Update api_key check logic * Properly support zipped dataset feature * update docstring * Revert tuorial change * extend changes to log_dataset * add run name * bug fix * bug fix * Update comment * fix import check * remove unused import * Hardcore .yaml file extension * reduce code * Reformat using pycharm * Remove redundant try catch * More refactoring and bug fixes * retry * Reformat using pycharm * respect LOGGERS include list * Fix * fix Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>pull/4405/head
parent
75d323dd8b
commit
a4e4553040
|
@ -157,6 +157,8 @@ class WandbLogger():
|
|||
self.data_dict = data_dict
|
||||
else: # Local .yaml dataset file or .zip file
|
||||
self.data_dict = check_dataset(opt.data)
|
||||
else:
|
||||
self.data_dict = check_dataset(opt.data)
|
||||
|
||||
self.setup_training(opt)
|
||||
if not self.wandb_artifact_data_dict:
|
||||
|
@ -505,4 +507,4 @@ def all_logging_disabled(highest_level=logging.CRITICAL):
|
|||
try:
|
||||
yield
|
||||
finally:
|
||||
logging.disable(previous_level)
|
||||
logging.disable(previous_level)
|
Loading…
Reference in New Issue