composer.trainer.devices.device_gpu#
The GPU device used for training.
Classes
- 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.