examples/research_projects/seq2seq-distillation/utils.py/AbstractSeq2SeqDataset/collate_fn
class AbstractSeq2SeqDataset:    def collate_fn(self, batch):
        raise NotImplementedError("You must implement this")