diff --git a/train.py b/train.py index 410f16fed3bf..ebe6c2e8f5f9 100644 --- a/train.py +++ b/train.py @@ -172,7 +172,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary optimizer.add_param_group({'params': g1, 'weight_decay': hyp['weight_decay']}) # add g1 with weight_decay optimizer.add_param_group({'params': g2}) # add g2 (biases) LOGGER.info(f"{colorstr('optimizer:')} {type(optimizer).__name__} with parameter groups " - f"{len(g0)} weight, {len(g1)} weight (no decay), {len(g2)} bias") + f"{len(g0)} weight (no decay), {len(g1)} weight, {len(g2)} bias") del g0, g1, g2 # Scheduler