Improves docs and handling of entities and resuming by WandbLogger (#3264)
* adds latest tag to match wandb defaults * adds entity handling, 'last' tag * fixes bug causing finished runs to resume * removes redundant "last" tag for wandb artifact
This commit is contained in:
1 parent
dd7f0b7e05
commit
19100ba007
2 files changed
+25
-10
No files matched your search
@@ -443,7 +443,7 @@ def train(hyp, opt, device, tb_writer=None):
|
||||
if wandb_logger.wandb and not opt.evolve: # Log the stripped model
|
||||
wandb_logger.wandb.log_artifact(str(final), type='model',
|
||||
name='run_' + wandb_logger.wandb_run.id + '_model',
|
||||
aliases=['last', 'best', 'stripped'])
|
||||
aliases=['latest', 'best', 'stripped'])
|
||||
wandb_logger.finish_run()
|
||||
else:
|
||||
dist.destroy_process_group()
|
||||
|
||||
Reference in new issue
Block a user