composer.core#

Central components used by other modules.

Central parts of composer such as Engine, base class for critical components such as Algorithm and Callback and other useful functionality such as Timestamp are implemented under core.

Functions

ensure_data_spec

Ensures that the dataloader is a DataSpec.

ensure_evaluator

Ensure that evaluator is an Evaluator.

ensure_time

Ensure maybe_time is an instance of Time.

get_precision_context

Returns a context manager to automatically cast to a specific precision.

Classes

MemoryFormat

Enum class to represent different memory formats.

Precision

Enum class for the numerical precision to be used by the model.

Serializable

Interface for serialization; used by checkpointing.

Trace

Record of an algorithm's execution.

TrainerMode

Enum to represent which mode the Trainer is in.

Exceptions

BreakEpochException

Raising this exception will immediately end the current epoch.