Python CIFAR10の例

コード例 #1

ファイル: test_dataloader.py プロジェクト: 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

コード例 #2

ファイル: test_datamodules.py プロジェクト: Borda/lightning-bolts

def test_dev_datasets(datadir):

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