Customized load.
(self, state_dict, strict=True)
| 1004 | return state_dict_ |
| 1005 | |
| 1006 | def load_state_dict(self, state_dict, strict=True): |
| 1007 | """Customized load.""" |
| 1008 | |
| 1009 | if self._language_model_key in state_dict: |
| 1010 | state_dict = state_dict[self._language_model_key] |
| 1011 | self.language_model.load_state_dict(state_dict, strict=strict) |
nothing calls this directly
no test coverage detected