Customized load.
(self, state_dict, strict=True)
| 106 | return state_dict_ |
| 107 | |
| 108 | def load_state_dict(self, state_dict, strict=True): |
| 109 | """Customized load.""" |
| 110 | |
| 111 | if self._language_model_key in state_dict: |
| 112 | state_dict = state_dict[self._language_model_key] |
| 113 | self.language_model.load_state_dict(state_dict, strict=strict) |
| 114 | |
| 115 | |
| 116 | def CrossEntropy(output, labels): |
no outgoing calls
no test coverage detected