composer.core.types#
Reference for common types used throughout the composer library.
- composer.core.types.Batch[source]#
Alias to type Any. A batch of data can be represented in several formats, depending on the application.
- Type
Any
- composer.core.types.PyTorchScheduler[source]#
Alias for base class of learning rate schedulers such as
torch.optim.lr_scheduler.ConstantLR
.- Type
torch.optim.lr_scheduler._LRScheduler
- composer.core.types.Dataset[source]#
Alias for
torch.utils.data.Dataset
.- Type
Dataset[Batch]
Classes
Enum class to represent different memory formats. |
|
Enum to represent which mode the Trainer is in. |
Exceptions
Raising this exception will immediately end the current epoch. |