composer.trainer.devices.device_gpu#

The GPU device used for training.

Classes

An extension of Device for GPUs.

class composer.trainer.devices.device_gpu.DeviceGPU[source]#

Bases: composer.trainer.devices.device.Device

An extension of Device for GPUs.

This class takes no arguments.