Customized load.
(self, state_dict, strict=True)
| 1095 | return state_dict_ |
| 1096 | |
| 1097 | def load_state_dict(self, state_dict, strict=True): |
| 1098 | """Customized load.""" |
| 1099 | |
| 1100 | if self._language_model_key in state_dict: |
| 1101 | state_dict = state_dict[self._language_model_key] |
| 1102 | self.language_model.load_state_dict(state_dict, strict=strict) |
nothing calls this directly
no test coverage detected