Commit
•
b73c62e
1
Parent(s):
4c40933
W&B: Remember batchsize on resuming (#6512)
Browse files* 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>
train.py
CHANGED
@@ -96,7 +96,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
|
|
96 |
if loggers.wandb:
|
97 |
data_dict = loggers.wandb.data_dict
|
98 |
if resume:
|
99 |
-
weights, epochs, hyp = opt.weights, opt.epochs, opt.hyp
|
100 |
|
101 |
# Register actions
|
102 |
for k in methods(loggers):
|
|
|
96 |
if loggers.wandb:
|
97 |
data_dict = loggers.wandb.data_dict
|
98 |
if resume:
|
99 |
+
weights, epochs, hyp, batch_size = opt.weights, opt.epochs, opt.hyp, opt.batch_size
|
100 |
|
101 |
# Register actions
|
102 |
for k in methods(loggers):
|