예제 #1
0
 def val_dataloader(self):
     return DataLoader(dataset=RandomClassData(100, 10),
                       batch_size=self.hparams['batch-size'],
                       num_workers=int(WORKERS))
예제 #2
0
 def train_dataloader(self):
     return DataLoader(dataset=RandomClassData(200, 10), batch_size=self.hparams['batch-size'], num_workers=N_CPUS)