def METHOD_NAME(self):
    self.train_loader = torch.utils.data.DataLoader(
        self.dataset1, **self.train_kwargs
    )
    self.test_loader = torch.utils.data.DataLoader(
        self.dataset2, **self.test_kwargs
    )
    self.next(self.train_model)