Config

jax_dataloader.utils.Config

Global configuration for the library

Source code in jax_dataloader/utils.py

@dataclass
class Config:
    """Global configuration for the library"""
    rng_reserve_size: int
    global_seed: int

    @classmethod
    def default(cls) -> Config:
        return cls(rng_reserve_size=1, global_seed=42)

Attributes

global_seed: int instance-attribute

rng_reserve_size: int instance-attribute