提交 0cc6bc0b 编写于 作者: T tianyi1997 提交者: HydrogenSulfate

Add regex support for selecting optim object

上级 8d4a79e5
......@@ -16,6 +16,7 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import re
import copy
import paddle
from typing import Dict, List
......@@ -120,6 +121,11 @@ def build_optimizer(config, epochs, step_each_epoch, model_list=None):
optim_model.append(model_list[i])
elif hasattr(model_list[i], optim_scope):
optim_model.append(getattr(model_list[i], optim_scope))
else:
for name, layer in model_list[i].named_sublayers():
if len(layer.parameters()) != 0 \
and re.fullmatch(optim_scope, name):
optim_model.append(layer)
optim = getattr(optimizer, optim_name)(
learning_rate=lr, grad_clip=grad_clip,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册