✨Enhanced before_summary

carefree0910 · Jan 20, 2025 · 87d765a · 87d765a
1 parent d9de2f8
commit 87d765a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/learn/callbacks/defaults.py b/core/learn/callbacks/defaults.py
@@ -71,8 +71,8 @@ def before_summary(self, trainer: ITrainer) -> None:
             exclude = finetune_config.get("exclude", "")
             exclude_names: List[str]
             if not exclude:
+                strict = finetune_config.get("strict", True)
                 exclude_names = []
-                model.load_state_dict(states)
             else:
                 exclude_names = []
                 for name in states.keys():
@@ -93,7 +93,7 @@ def before_summary(self, trainer: ITrainer) -> None:
                         )
                     states = {k: v for k, v in states.items() if k not in exclude_names}
                     strict = finetune_config.get("strict", False)
-                model.load_state_dict(states, strict=strict)
+            model.load_state_dict(states, strict=strict)
             freeze = finetune_config.get("freeze", "")
             freeze_except = finetune_config.get("freeze_except", "")
             if freeze or freeze_except: