From f1d73a29e549654c99674bf07dd8f7a2f5c19d18 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 30 Apr 2020 12:26:02 -0700 Subject: [PATCH] Optimizer group report --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index bf9d6082..67c20aa3 100644 --- a/train.py +++ b/train.py @@ -108,6 +108,7 @@ def train(): optimizer = optim.SGD(pg0, lr=hyp['lr0'], momentum=hyp['momentum'], nesterov=True) optimizer.add_param_group({'params': pg1, 'weight_decay': hyp['weight_decay']}) # add pg1 with weight_decay optimizer.add_param_group({'params': pg2}) # add pg2 (biases) + print('Optimizer groups: %g .bias, %g Conv2d.weight, %g other' % (len(pg2), len(pg1), len(pg0))) del pg0, pg1, pg2 start_epoch = 0