def _get_state_dict(self):
    model_state = {'state_dict': self.state_dict(), 'current_task': self._current_task, 'tag_type': self.get_current_label_type(), 'tag_dictionary': self.get_current_label_dictionary(), 'tars_model': self.tars_model, 'num_negative_labels_to_sample': self.num_negative_labels_to_sample, 'prefix': self.prefix, 'task_specific_attributes': self._task_specific_attributes}
    return model_state
def _get_state_dict(self):
    model_state = {'state_dict': self.state_dict(), 'current_task': self._current_task, 'tag_type': self.get_current_label_type(), 'tag_dictionary': self.get_current_label_dictionary(), 'tars_model': self.tars_model, 'num_negative_labels_to_sample': self.num_negative_labels_to_sample, 'prefix': self.prefix, 'task_specific_attributes': self._task_specific_attributes}
    return model_state