openfl.component.collaborator.collaborator.DevicePolicy

openfl.component.collaborator.collaborator.DevicePolicy#

class openfl.component.collaborator.collaborator.DevicePolicy(value)[source]#

Bases: Enum

Device assignment policy.

CPU_ONLY#

Assigns tasks to CPU only.

Type:

int

CUDA_PREFERRED#

Prefers CUDA for task assignment if available.

Type:

int

__init__()#

Attributes