未验证 提交 35b1c4a4 编写于 作者: D Double_V 提交者: GitHub

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
上级 95e07d18
...@@ -342,7 +342,7 @@ class OptimizerBuilder(): ...@@ -342,7 +342,7 @@ class OptimizerBuilder():
n: p n: p
for n, p in model.named_parameters() for n, p in model.named_parameters()
if any([k in n 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 = group.copy()
_group.update({'params': list(_params.values())}) _group.update({'params': list(_params.values())})
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册