fix code bug
parent
aa26a8c1d8
commit
0146af28b0
|
@ -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...
|
||||
|
|
Loading…
Reference in New Issue