Ayush Chaurasia glenn-jocher commited on
Commit
cd540d8
1 Parent(s): b74929c

W&B: suppress warnings (#4257)

Browse files

* 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

* call wandblogger.log instead of wandb.log

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Files changed (1) hide show
  1. utils/loggers/__init__.py +2 -1
utils/loggers/__init__.py CHANGED
@@ -134,7 +134,8 @@ class Loggers():
134
  self.tb.add_image(f.stem, np.asarray(Image.open(f)), epoch, dataformats='HWC')
135
 
136
  if self.wandb:
137
- wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]})
 
138
  wandb.log_artifact(str(best if best.exists() else last), type='model',
139
  name='run_' + self.wandb.wandb_run.id + '_model',
140
  aliases=['latest', 'best', 'stripped'])
 
134
  self.tb.add_image(f.stem, np.asarray(Image.open(f)), epoch, dataformats='HWC')
135
 
136
  if self.wandb:
137
+ self.wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]})
138
+ # Calling wandb.log. TODO: Refactor this into WandbLogger.log_model
139
  wandb.log_artifact(str(best if best.exists() else last), type='model',
140
  name='run_' + self.wandb.wandb_run.id + '_model',
141
  aliases=['latest', 'best', 'stripped'])