Unverified 提交 2acbe969 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

W&B epoch logging update (#2073)

上级 630ec06a
......@@ -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]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论