def test_dataloader(self): return DataLoader(RandomDataset(32, 64), batch_size=2)
def val_dataloader(): return DataLoader(RandomDataset(32, 64), batch_size=2)