Unverified 提交 b73c62eb authored 作者: Ayush Chaurasia's avatar Ayush Chaurasia 提交者: GitHub

W&B: Remember batchsize on resuming (#6512)

* log best.pt metrics at train end * update * Update __init__.py * flush callbacks when using evolve * remember batch size on resuming * Update train.py Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 4c409332
...@@ -96,7 +96,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary ...@@ -96,7 +96,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
if loggers.wandb: if loggers.wandb:
data_dict = loggers.wandb.data_dict data_dict = loggers.wandb.data_dict
if resume: if resume:
weights, epochs, hyp = opt.weights, opt.epochs, opt.hyp weights, epochs, hyp, batch_size = opt.weights, opt.epochs, opt.hyp, opt.batch_size
# Register actions # Register actions
for k in methods(loggers): for k in methods(loggers):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论