fix code bug

pull/1868/head
HydrogenSulfate 2022-04-21 17:11:07 +08:00
parent aa26a8c1d8
commit 0146af28b0
1 changed files with 1 additions and 1 deletions

View File

@ -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...