composer.trainer.devices.device_hparams#

The Hparams used to construct devices.

Hparams

These classes are used with yahp for YAML-based configuration.

CPUDeviceHparams

Used to construct a DeviceCPU

DeviceHparams

Base for CPUDeviceHparams and GPUDeviceHparams

GPUDeviceHparams

Used to construct a DeviceGPU
class composer.trainer.devices.device_hparams.CPUDeviceHparams[source]#

Bases: composer.trainer.devices.device_hparams.DeviceHparams

Used to construct a DeviceCPU

class composer.trainer.devices.device_hparams.DeviceHparams[source]#

Bases: yahp.hparams.Hparams

Base for CPUDeviceHparams and GPUDeviceHparams

class composer.trainer.devices.device_hparams.GPUDeviceHparams[source]#

Bases: composer.trainer.devices.device_hparams.DeviceHparams

Used to construct a DeviceGPU