W&B epoch logging update (#2073)
Browse files
train.py
CHANGED
@@ -364,7 +364,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
|
|
364 |
if tb_writer:
|
365 |
tb_writer.add_scalar(tag, x, epoch) # tensorboard
|
366 |
if wandb:
|
367 |
-
wandb.log({tag: x}) # W&B
|
368 |
|
369 |
# Update best mAP
|
370 |
fi = fitness(np.array(results).reshape(1, -1)) # weighted combination of [P, R, [email protected], [email protected]]
|
|
|
364 |
if tb_writer:
|
365 |
tb_writer.add_scalar(tag, x, epoch) # tensorboard
|
366 |
if wandb:
|
367 |
+
wandb.log({tag: x}, step=epoch, commit=tag == tags[-1]) # W&B
|
368 |
|
369 |
# Update best mAP
|
370 |
fi = fitness(np.array(results).reshape(1, -1)) # weighted combination of [P, R, [email protected], [email protected]]
|