Ayush Chaurasia
GitHub
3 years ago
2 changed files with 9 additions and 5 deletions
+ 8
- 4
utils/loggers/__init__.py
View File
| if self.wandb: | if self.wandb: | ||||
| self.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 | # 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']) | |||||
| self.wandb.finish_run() | |||||
| if not self.opt.evolve: | |||||
| wandb.log_artifact(str(best if best.exists() else last), type='model', | |||||
| name='run_' + self.wandb.wandb_run.id + '_model', | |||||
| aliases=['latest', 'best', 'stripped']) | |||||
| self.wandb.finish_run() | |||||
| else: | |||||
| self.wandb.finish_run() | |||||
| self.wandb = WandbLogger(self.opt) |
+ 1
- 1
utils/loggers/wandb/wandb_utils.py
View File
| https://docs.wandb.com/guides/integrations/yolov5 | https://docs.wandb.com/guides/integrations/yolov5 | ||||
| """ | """ | ||||
| def __init__(self, opt, run_id, job_type='Training'): | |||||
| def __init__(self, opt, run_id=None, job_type='Training'): | |||||
| """ | """ | ||||
| - Initialize WandbLogger instance | - Initialize WandbLogger instance | ||||
| - Upload dataset if opt.upload_dataset is True | - Upload dataset if opt.upload_dataset is True |
Loading…