Ejemplos de RandomClassData en Python

Ejemplo n.º 1

Archivo: module.py Proyecto: mRcSchwering/pytorch_lightning_test

 def val_dataloader(self):
     return DataLoader(dataset=RandomClassData(100, 10),
                       batch_size=self.hparams['batch-size'],
                       num_workers=int(WORKERS))

Ejemplo n.º 2

Archivo: module.py Proyecto: mRcSchwering/pytorch_lightning_test

 def train_dataloader(self):
     return DataLoader(dataset=RandomClassData(200, 10), batch_size=self.hparams['batch-size'], num_workers=N_CPUS)