composer.trainer.devices

Modules

composer.trainer.devices.device	The base Device class.
composer.trainer.devices.device_cpu	The CPU device used for training.
composer.trainer.devices.device_gpu	The GPU device used for training.
composer.trainer.devices.device_hparams	The Hparams used to construct devices.

Module for devices on which models run.

Used by Trainer in order to train on different devices such as GPU and CPU.

Classes

Device	Abstract class for a device on which a model runs.
DeviceCPU	An extension of Device for CPUs.
DeviceGPU	An extension of Device for GPUs.