openfl.component.collaborator.collaborator.DevicePolicy

class openfl.component.collaborator.collaborator.DevicePolicy(value)

Bases: Enum

Device assignment policy.

Class Attributes:

• CPU_ONLY (int) – Assigns tasks to CPU only.

• CUDA_PREFERRED (int) – Prefers CUDA for task assignment if available.

Attributes

CPU_ONLY
CUDA_PREFERRED