(self, state_dict)
| 519 | ) |
| 520 | |
| 521 | def load_state_dict(self, state_dict): |
| 522 | if 'config' in state_dict: |
| 523 | self.config.update(ConfigDict(state_dict['config'])) |
| 524 | self._index = state_dict.get('index', self.config.example_index_at_start) |
| 525 | self._file_loc = state_dict.get('file_loc', self.config.start_seek_loc) |
| 526 | self._total_tokens = state_dict.get('total_tokens', self.config.tokens_count_at_start) |
| 527 | |
| 528 | @property |
| 529 | def seq_length(self): |