green_ai_bench.cpu_inference

source module green_ai_bench.cpu_inference

CPU inference script.

Classes

  • CPUInference — CPU-specific implementation of BaseInference for model inference on CPU hardware.

source class CPUInference(model_path, use_GPU=False, **kwargs)

Bases : BaseInference

CPU-specific implementation of BaseInference for model inference on CPU hardware.

Initialize the CPUInference class.

Methods

  • infer — Run inference on the model.

source method CPUInference.infer()

Run inference on the model.

Raises

  • RuntimeError