Skip to content

strategy

strategy

Data loading for statically declared dataset mixtures.

Sampling(dataset: type[DatasetComponent], rate: float, style: DatasetStyleLike = DatasetStyle.PADDED) dataclass

Weight and reader style for one statically declared dataset.

Strategy(spec: ExecutionSpec, block_size: int, mixture: list[Sampling])

config(mixture: list[Sampling]) -> list[type[Any]] staticmethod

Return configuration schemas required by a dataset mixture.

Parameters:

Name Type Description Default
mixture list[Sampling]

Static dataset mixture declared by a trainer.

required

Returns:

Type Description
list[type[Any]]

Dataset configuration schemas in declaration order.