From db1f83be6312a3d68c817fd25251194e5e7e5b5d Mon Sep 17 00:00:00 2001 From: Otfot Date: Mon, 17 Jan 2022 14:58:34 +0800 Subject: [PATCH] Fix `train.py` parameter groups desc error (#6318) * Fix `train.py` parameter groups desc error * Cleanup Co-authored-by: Glenn Jocher --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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