Ejemplos de CIFAR10 en Python

Ejemplo n.º 1

Archivo: test_dataloader.py Proyecto: supadupa/pytorch-lightning-bolts

def test_async_dataloader(tmpdir):
    ds = CIFAR10(tmpdir)

    if torch.cuda.device_count() > 0:  # Can only run this test with a GPU
        device = torch.device('cuda', 0)
        dataloader = AsynchronousLoader(ds, device=device)

        for b in dataloader:
            pass

        dataloader = AsynchronousLoader(DataLoader(ds, batch_size=16),
                                        device=device)
        for b in dataloader:
            pass

Ejemplo n.º 2

Archivo: test_datamodules.py Proyecto: Borda/lightning-bolts

def test_dev_datasets(datadir):

    ds = CIFAR10(data_dir=datadir)
    for _ in ds:
        pass