composer.algorithms.channels_last.channels_last#
composer.algorithms.channels_last.channels_last
Functions
Changes the memory format of the model to torch.channels_last. |
Classes
Base class for algorithms. |
|
Changes the memory format of the model to |
|
Enum to represent events in the training loop. |
|
An interface to record training data. |
|
The state of the trainer. |
Attributes
Optional
annotations
log
- class composer.algorithms.channels_last.channels_last.ChannelsLast[source]#
Bases:
composer.core.algorithm.Algorithm
Changes the memory format of the model to
torch.channels_last
. This usually yields improved GPU utilization.Runs on
Event.INIT
, so it can set the memory format before the model is DDP wrapped. Has no hyperparameters.