optimizer的参数错误修正

paradoxjun · Oct 10, 2021 · af80609 · af80609
1 parent 83e9c45
commit af80609
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/rec_train.py b/tools/rec_train.py
@@ -312,7 +312,7 @@ def main():
     # ===> get fine tune layers
     params_to_train = get_fine_tune_params(net, train_options['fine_tune_stage'])
     # ===> solver and lr scheduler
-    optimizer = build_optimizer(net.parameters(), cfg['optimizer'])
+    optimizer = build_optimizer(params_to_train, cfg['optimizer'])
     scheduler = build_scheduler(optimizer, cfg['lr_scheduler'])
 
     # ===> whether to resume from checkpoint