Skip to content

Commit

Permalink
W&B: Remember batchsize on resuming (ultralytics#6512)
Browse files Browse the repository at this point in the history
* 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>
  • Loading branch information
AyushExel and glenn-jocher committed Feb 3, 2022
1 parent 92f5b8b commit 1c7f721
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 1c7f721

Please sign in to comment.