* 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>modifyDataloader
@@ -134,7 +134,8 @@ class Loggers(): | |||
self.tb.add_image(f.stem, np.asarray(Image.open(f)), epoch, dataformats='HWC') | |||
if self.wandb: | |||
wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]}) | |||
self.wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]}) | |||
# Calling wandb.log. TODO: Refactor this into WandbLogger.log_model | |||
wandb.log_artifact(str(best if best.exists() else last), type='model', | |||
name='run_' + self.wandb.wandb_run.id + '_model', | |||
aliases=['latest', 'best', 'stripped']) |