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.
validate_eval_automicrobatching	Ensure automicrobatching is only on GPU.

Classes

MemoryFormat	Enum class to represent different memory formats.
Precision	Enum class for the numerical precision to be used by the model.
PyTorchScheduler	alias of torch.optim.lr_scheduler.LRScheduler
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.