examples/research_projects/seq2seq-distillation/lightning_base.py/BaseTransformer/test_step
class BaseTransformer:    def test_step(self, batch, batch_nb):
        return self.validation_step(batch, batch_nb)