W&B epoch logging update (#2073)
This commit is contained in:
1 parent
630ec06af6
commit
2acbe9699a
1 file changed
+1
-1
@@ -364,7 +364,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
|
||||
if tb_writer:
|
||||
tb_writer.add_scalar(tag, x, epoch) # tensorboard
|
||||
if wandb:
|
||||
wandb.log({tag: x}) # W&B
|
||||
wandb.log({tag: x}, step=epoch, commit=tag == tags[-1]) # W&B
|
||||
|
||||
# Update best mAP
|
||||
fi = fitness(np.array(results).reshape(1, -1)) # weighted combination of [P, R, mAP@.5, mAP@.5-.95]
|
||||
|
||||
Reference in new issue
Block a user