W&B: More improvements and refactoring (#4205)
* 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 Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
e016b15555
commit
750465edae
2 files changed
+8
-7
No files matched your search
@@ -48,12 +48,12 @@ class Loggers():
|
||||
self.tb = SummaryWriter(str(s))
|
||||
|
||||
# W&B
|
||||
try:
|
||||
assert 'wandb' in self.include and wandb
|
||||
run_id = torch.load(self.weights).get('wandb_id') if self.opt.resume else None
|
||||
if wandb and 'wandb' in self.include:
|
||||
wandb_artifact_resume = isinstance(self.opt.resume, str) and self.opt.resume.startswith('wandb-artifact://')
|
||||
run_id = torch.load(self.weights).get('wandb_id') if self.opt.resume and not wandb_artifact_resume else None
|
||||
self.opt.hyp = self.hyp # add hyperparameters
|
||||
self.wandb = WandbLogger(self.opt, run_id)
|
||||
except:
|
||||
else:
|
||||
self.wandb = None
|
||||
|
||||
return self
|
||||
|
||||
Reference in new issue
Block a user