Skip to content

repeat

Dataset wrapper that lengthens an epoch by repeating an underlying dataset.

Classes:

  • RepeatDataset –

    Repeats a dataset loop times so one epoch iterates it loop times.

RepeatDataset 

RepeatDataset(dataset: Dataset, loop: int)

Bases: Dataset

Repeats a dataset loop times so one epoch iterates it loop times.

Lengthens an epoch (and thus the number of optimizer steps per epoch) without otherwise changing training.

Parameters:

  • dataset (Dataset) –

    The dataset to repeat.

  • loop (int) –

    Number of times the dataset is iterated per epoch.