def __getitem__(self, idx):
    item = {key: torch.tensor(val[idx]) for (key, val) in self.encodings.items()}
    item['labels'] = item['input_ids']
    return item
def __getitem__(self, idx):
    item = {key: torch.tensor(val[idx]) for (key, val) in self.encodings.items()}
    item['labels'] = item['input_ids']
    return item