Ayush Chaurasia glenn-jocher commited on
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>

Files changed (1) hide show
  1. train.py +1 -1
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):