Kathy Carter
Results
3
issues of
Kathy Carter
script/main.py中 class DataCollatorForLMDataset(object): def __call__(self, instances: Sequence[Dict]) -> Dict[str, torch.Tensor]: input_ids, labels = tuple([instance[key].unsqueeze(0) for instance in instances] for key in ("input_ids", "labels")) input_ids = torch.cat(input_ids, dim=0) labels = torch.cat(labels,...