mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

W&B: Don't log models in evolve operation (#4611)

Browse Source
This commit is contained in:
Ayush Chaurasia 2021-08-30 20:37:20 +05:30 committed by GitHub
parent 3a72d4a7e3
commit 11e7c7b48d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
utils/loggers/__init__.py
View File

@ -138,7 +138,11 @@ class Loggers():
if self.wandb:
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'])
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)

2
utils/loggers/wandb/wandb_utils.py
View File

@ -112,7 +112,7 @@ class WandbLogger():
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
- Upload dataset if opt.upload_dataset is True