primeqa.qg.utils.data_collator.T2TDataCollator

class primeqa.qg.utils.data_collator.T2TDataCollator

Bases: object

Methods

__call__(batch: List) → Dict[str, torch.Tensor]

Take a list of samples from a Dataset and collate them into a batch. :returns: A dictionary of tensors