I rather have it default to non-strict loading instead so I can clean up YAMLs

This commit is contained in:
mrq 2024-07-30 22:24:38 -05:00
parent d7c6be6f78
commit 52d13b321f

View File

@ -593,7 +593,7 @@ class Trainer:
load_state_dict: bool = False load_state_dict: bool = False
load_states: bool = True load_states: bool = True
strict_loading: bool = True strict_loading: bool = False
load_module_only: bool = False load_module_only: bool = False
restart_step_count: bool = False restart_step_count: bool = False