composer.core.types#

Reference for common types used throughout the composer library.

composer.core.types.Batch#

Alias to type Any. A batch of data can be represented in several formats, depending on the application.

Type

Any

composer.core.types.JSON#

JSON Data.

Type

str | float | int | None | List[โ€™JSONโ€™] | Dict[str, โ€™JSONโ€™]

composer.core.types.Dataset[source]#

Alias for torch.utils.data.Dataset.

Type

Dataset[Batch]

Classes

MemoryFormat

Enum class to represent different memory formats.

TrainerMode

Enum to represent which mode the Trainer is in.