examples/research_projects/seq2seq-distillation/finetune.py/SummarizationModule/test_step
class SummarizationModule:    def test_step(self, batch, batch_idx):
        return self._generative_step(batch)