docs.unity3d.com
Search Results for

    Show / Hide Table of Contents

    Create an engine to run a model

    To run a model, create a worker. A worker is the engine that breaks the model down into executable tasks and schedules the tasks to run on a back end (typically the GPU or CPU).

    A worker is an instance of an IWorker object.

    Create a Worker

    Use WorkerFactory.CreateWorker to create a worker. You must specify a back end type, which tells Sentis where to run the worker, and a runtime model.

    For example, the following code creates a worker that runs on the GPU using Sentis compute shaders.

    using UnityEngine;
    using Unity.Sentis;
    
    public class CreateWorker : MonoBehaviour
    {
        ModelAsset modelAsset;
        Model runtimeModel;
        IWorker worker;
        
        void Start()
        {
            runtimeModel = ModelLoader.Load(modelAsset);
            worker = WorkerFactory.CreateWorker(BackendType.GPUCompute, runtimeModel);
        }
    }
    

    Back end types

    Sentis provides CPU and GPU back end types. To understand how Sentis executes operations using the different back ends, refer to How Sentis runs a model.

    If a back end type doesn't support a Sentis layer in a model, the worker will assert. Refer to Supported ONNX operators for more information.

    BackendType.GPUCompute, BackendType.GPUCommandBuffer and BackendType.CPU are the fastest back end types, so use BackendType.GPUPixel only if a platform doesn't support compute shaders. To check if your runtime platform supports compute shaders use SystemInfo.supportsComputeShaders

    If you use BackendType.CPU with WebGL, Burst compiles to WebAssembly code which might be slow. Refer to Getting started with WebGL development for more information.

    How fast a model runs depends on how well a platform supports multithreading for Burst, or how fully it supports compute shaders. You can profile a model to understand the performance of a model.

    Additional resources

    • Create a runtime model
    • How Sentis runs a model
    • Supported ONNX operators
    • Run a model
    In This Article
    • Create a Worker
    • Back end types
    • Additional resources
    Back to top
    Copyright © 2024 Unity Technologies — Trademarks and terms of use
    • Legal
    • Privacy Policy
    • Cookie Policy
    • Do Not Sell or Share My Personal Information
    • Your Privacy Choices (Cookie Settings)