From a4e4553040503b11df9283fda666736e9c57dd87 Mon Sep 17 00:00:00 2001 From: Ayush Chaurasia Date: Thu, 12 Aug 2021 01:26:25 +0530 Subject: [PATCH] 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 --- utils/loggers/wandb/wandb_utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/utils/loggers/wandb/wandb_utils.py b/utils/loggers/wandb/wandb_utils.py index 66fa8f85ec4e..3f2684a7f3e3 100644 --- a/utils/loggers/wandb/wandb_utils.py +++ b/utils/loggers/wandb/wandb_utils.py @@ -157,6 +157,8 @@ def __init__(self, opt, run_id, job_type='Training'): 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) \ No newline at end of file