if self._dataset_path == 'wikitext': self._dataset_name = 'wikitext-2-raw-v1' data = load_dataset(self._dataset_path, self._dataset_name, split=self._split) elif self._dataset_path == 'c4': self._dataset_name = 'realnewslike' data = load_dataset(self._dataset_path, self....