unilab.algos.torch.common.ane_inference.ANEInference

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

Bases: object

ANE inference wrapper.

Methods

__init__(coreml_model)

predict(obs_np)

Run inference.

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

Run inference.