* 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>modifyDataloader
@@ -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) |