composer.trainer.devices

Devices

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.

Device Dataloaders

CudaDataLoader

Wraps DataLoader and moves samples onto the specified device as they are used.