glenn-jocher commited on
Commit
2acbe96
·
unverified ·
1 Parent(s): 630ec06

W&B epoch logging update (#2073)

Browse files
Files changed (1) hide show
  1. train.py +1 -1
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]]