unilab.algos.torch.common.ane_inference

Simplified ANE backend using deterministic inference.

Functions

create_ane_actor(actor_model, obs_dim, ...)

Create ANE-compatible actor using deterministic inference.

Classes

ANEInference

ANE inference wrapper.
unilab.algos.torch.common.ane_inference.create_ane_actor(actor_model, obs_dim, action_dim)[source]

Create ANE-compatible actor using deterministic inference.

Uses mean action (no sampling) to avoid CoreML limitations.

class unilab.algos.torch.common.ane_inference.ANEInference[source]

Bases: object

ANE inference wrapper.

__init__(coreml_model)[source]
predict(obs_np)[source]

Run inference.