Skip to content

Select the training device according to configuration #5136

@leoll2

Description

@leoll2

After #5115, the training job has an input parameter device to specify the training device.

To make this parameter actually work, the OTXEngine device parameter must be initialized accordingly:

device=DeviceType.cpu, # TODO make device configurable (#5040)

Within this ticket, make sure the training runs on the selected device.

Metadata

Metadata

Assignees

Labels

Geti Tune BackendIssues related to Geti Tune backend

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions