cascade.data.DataModule

class cascade.data.DataModule(batch_size, pin_memory, val_frac, random_state)[source]

Bases: LightningDataModule

Abstract data module

Parameters:

  • batch_size (int) – Batch size

  • pin_memory (bool) – Whether to use pin memory

  • val_frac (float) – Fraction of validation data

  • random_state (RandomState | int | None) – Random state