提交 0146af28 编写于 作者: H HydrogenSulfate

fix code bug

上级 aa26a8c1
......@@ -111,7 +111,7 @@ def build_optimizer(config, epochs, step_each_epoch, model_list=None):
if optim_scope.endswith("Loss"):
# optimizer for loss
for m in model_list[i].sublayers(True):
if m.__class_name == optim_scope:
if m.__class__.__name__ == optim_scope:
optim_model.append(m)
else:
# opmizer for module in model, such as backbone, neck, head...
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册