composer.trainer.devices.device_hparams#
The Hparams
used to construct devices.
Hparams
These classes are used with yahp
for YAML
-based configuration.
Used to construct a |
|
Base for |
|
Used to construct a |
- 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
andGPUDeviceHparams
- class composer.trainer.devices.device_hparams.GPUDeviceHparams[source]#
Bases:
composer.trainer.devices.device_hparams.DeviceHparams
Used to construct a
DeviceGPU