transformer_rankers.datasets.dataset.collate_T2T_batch

transformer_rankers.datasets.dataset.collate_T2T_batch(batch: List) → Dict[str, torch.Tensor]

Data collator for generation tasks.