unreal.NeuralDeviceType

class unreal.NeuralDeviceType

Bases: unreal.EnumBase

It defines in which device (CPU, GPU) the desired operation (e.g., the neural network inference) is run.

C++ Source:

  • Plugin: NeuralNetworkInference

  • Module: NeuralNetworkInference

  • File: NeuralEnumClasses.h

CPU

The operation will occur on the CPU.

Type

0

GPU

The operation will occur on the GPU.

Type

1