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>

W&B: Remember batchsize on resuming (#6512)
b73c62eb · Ayush Chaurasia · GitHub · 4c409332 · b73c62eb
--- a/train.py
+++ b/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):