bugfix(resume): fix wrong lr_schedule

ZombaSY · Apr 22, 2022 · 08a6d48 · 08a6d48
1 parent 623b5f0
commit 08a6d48
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/train_semi.py b/train_semi.py
@@ -153,8 +153,9 @@ def main():
         load_state(cfg["saver"]["pretrain"], model, key="model_state")
         load_state(cfg["saver"]["pretrain"], model_teacher, key="teacher_state")
 
+    optimizer_start = get_optimizer(params_list, cfg_optim)
     lr_scheduler = get_scheduler(
-        cfg_trainer, len(train_loader_sup), optimizer, start_epoch=last_epoch
+        cfg_trainer, len(train_loader_sup), optimizer_start, start_epoch=last_epoch
     )
 
     # build class-wise memory bank