Unverified 提交 dc51e80b authored 作者: Ayush Chaurasia's avatar Ayush Chaurasia 提交者: GitHub

Fix: evolve with wandb (#2634)

上级 6e8c5b76
...@@ -439,7 +439,7 @@ def train(hyp, opt, device, tb_writer=None): ...@@ -439,7 +439,7 @@ def train(hyp, opt, device, tb_writer=None):
strip_optimizer(f) # strip optimizers strip_optimizer(f) # strip optimizers
if opt.bucket: if opt.bucket:
os.system(f'gsutil cp {final} gs://{opt.bucket}/weights') # upload os.system(f'gsutil cp {final} gs://{opt.bucket}/weights') # upload
if wandb_logger.wandb: # Log the stripped model if wandb_logger.wandb and not opt.evolve: # Log the stripped model
wandb_logger.wandb.log_artifact(str(final), type='model', wandb_logger.wandb.log_artifact(str(final), type='model',
name='run_' + wandb_logger.wandb_run.id + '_model', name='run_' + wandb_logger.wandb_run.id + '_model',
aliases=['last', 'best', 'stripped']) aliases=['last', 'best', 'stripped'])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论