Created by: ArmenAg
Patch Description Fix a small nit in the load_state_dict refactor code
Testing steps I ran training, that loaded checkpoint with this small change turned on. Everything worked.
Created by: ArmenAg
Patch Description Fix a small nit in the load_state_dict refactor code
Testing steps I ran training, that loaded checkpoint with this small change turned on. Everything worked.