Skip to content
heliaEDGE
Reference
HELIA

mlperf_tiny

Faithful scale-1 MLPerf Tiny architectures with newly initialized weights.

Pinned topology and licensing: https://ambiqai.github.io/helia-edge/examples/mlperf-tiny/ and models/licenses/. Adapted from MLCommons Tiny (Apache-2.0); anomaly detector originally Copyright (c) 2020 Hitachi, Ltd. (MIT). Changes: Keras 3 API, fixed captured shapes/configurations, explicit layer names, no training or weight downloads. These constructors do not provide official trained weights or accuracy claims.

Machine-readable model

  • buildfunctionBuild a faithful MLPerf Tiny architecture without loading or converting weights.
function

build

Python

Build a faithful MLPerf Tiny architecture without loading or converting weights.

helia_edge/models/mlperf_tiny.py:187

build(
params: MlperfTinyParams,
input_shape: tuple[int | None, ...] | None = None,
*,
batch_size: int | None = None,
name: str | None = None,
) -> keras.Model

Build a faithful MLPerf Tiny architecture without loading or converting weights.

Parameters of build
NameTypeDefaultDescription
paramsMlperfTinyParamsRequiredThe architecture: ``kws``, ``vww``, ``resnet`` or ``ad``.
input_shapetuple[int | None, ...] | NoneNoneNone, or the architecture's fixed input shape.
batch_sizeint | NoneNoneStatic batch size; None for a dynamic batch.
namestr | NoneNoneModel name; the family when None.
Returns of build
TypeDescription
keras.Modelkeras.Model: The model, named ``mlperf_tiny`` unless ``name`` is given.