W&B log epoch (#1946)
* W&B log epoch * capitalize * W&B log epoch * capitalize * Update train.py New try using https://docs.wandb.ai/library/log#incremental-logging * Update train.py * Update test.py * Update train.py * Update plots.py * Update train.py * Update train.py * label plot step -1 * update * update * update * update * update * update * Update train.py * Update train.py
This commit is contained in:
1 parent
08d3119e09
commit
59c21c7bcb
3 files changed
+4
-4
No files matched your search
@@ -321,7 +321,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
|
||||
# tb_writer.add_graph(model, imgs) # add model to tensorboard
|
||||
elif plots and ni == 10 and wandb:
|
||||
wandb.log({"Mosaics": [wandb.Image(str(x), caption=x.name) for x in save_dir.glob('train*.jpg')
|
||||
if x.exists()]})
|
||||
if x.exists()]}, commit=False)
|
||||
|
||||
# end batch ------------------------------------------------------------------------------------------------
|
||||
# end epoch ----------------------------------------------------------------------------------------------------
|
||||
|
||||
Reference in new issue
Block a user