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: Glenn Jocher <glenn.jocher@ultralytics.com>pull/6525/head
parent
4c40933266
commit
b73c62ebc5
2
train.py
2
train.py
|
@ -96,7 +96,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
|
|||
if loggers.wandb:
|
||||
data_dict = loggers.wandb.data_dict
|
||||
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
|
||||
for k in methods(loggers):
|
||||
|
|
Loading…
Reference in New Issue