BaseDataLoader

jax_dataloader.loaders.base.BaseDataLoader

Dataloader Interface

Source code in jax_dataloader/loaders/base.py

class BaseDataLoader:
    """Dataloader Interface"""

    def __init__(
        self, 
        dataset, 
        batch_size: int = 1,  # batch size
        shuffle: bool = False,  # if true, dataloader shuffles before sampling each batch
        num_workers: int = 0,  # how many subprocesses to use for data loading.
        drop_last: bool = False,
        generator: Optional[GeneratorType] = None,
        **kwargs
    ):
        pass

    def __len__(self):
        raise NotImplementedError

    def __next__(self):
        raise NotImplementedError

    def __iter__(self):
        raise NotImplementedError

Functions

__init__(dataset, batch_size: int = 1, shuffle: bool = False, num_workers: int = 0, drop_last: bool = False, generator: Optional[GeneratorType] = None, **kwargs)

Source code in jax_dataloader/loaders/base.py

def __init__(
    self, 
    dataset, 
    batch_size: int = 1,  # batch size
    shuffle: bool = False,  # if true, dataloader shuffles before sampling each batch
    num_workers: int = 0,  # how many subprocesses to use for data loading.
    drop_last: bool = False,
    generator: Optional[GeneratorType] = None,
    **kwargs
):
    pass

__iter__()

Source code in jax_dataloader/loaders/base.py

def __iter__(self):
    raise NotImplementedError

__len__()

Source code in jax_dataloader/loaders/base.py

def __len__(self):
    raise NotImplementedError

__next__()

Source code in jax_dataloader/loaders/base.py

def __next__(self):
    raise NotImplementedError