bugfix(train): fix typo

ZombaSY · Apr 8, 2022 · 623b5f0 · 623b5f0
1 parent 2b259fc
commit 623b5f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_semi.py b/train_semi.py
@@ -150,7 +150,7 @@ def main():
             )
 
     elif cfg["saver"].get("pretrain", False):
-        laod_state(cfg["saver"]["pretrain"], model, key="model_state")
+        load_state(cfg["saver"]["pretrain"], model, key="model_state")
         load_state(cfg["saver"]["pretrain"], model_teacher, key="teacher_state")
 
     lr_scheduler = get_scheduler(