From 35b1c4a4e52bd0416725984a1c360d20456f2c29 Mon Sep 17 00:00:00 2001 From: Double_V Date: Mon, 4 Jul 2022 19:48:20 +0800 Subject: [PATCH] fix params groups.trainable (#6354) * add FGD distill code * add configs * add doc * fix pretrain * pre-commit * fix ci * fix readme * fix readme * fix ci * fix param groups * fix * fix doc * fix doc,test=document_fix * fix params groups --- ppdet/optimizer/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/optimizer/optimizer.py b/ppdet/optimizer/optimizer.py index e6d5fe44a..e8a0dd8c8 100644 --- a/ppdet/optimizer/optimizer.py +++ b/ppdet/optimizer/optimizer.py @@ -342,7 +342,7 @@ class OptimizerBuilder(): n: p for n, p in model.named_parameters() if any([k in n - for k in group['params']] and p.trainable is True) + for k in group['params']]) and p.trainable is True } _group = group.copy() _group.update({'params': list(_params.values())}) -- GitLab