{
  "$schema": "https://ambiqai.github.io/helia-ui/schema/reference-model-1.json",
  "generatedFrom": {
    "sourceCommit": "f341fb11f7f5d77a4974ba8273c4cd55d67c21f0",
    "tool": "pyref",
    "version": "1.7.3"
  },
  "language": "python",
  "modules": [
    {
      "description": "Typed MBConv block parameters; importable without Keras.",
      "name": "mbconv_params",
      "path": "helia_edge.layers.mbconv_params",
      "submodules": [],
      "summary": "Typed MBConv block parameters; importable without Keras.",
      "symbols": [
        {
          "description": "MBConv parameters",
          "examples": [],
          "id": "helia_edge.layers.mbconv_params.MBConvParams",
          "kind": "class",
          "language": "python",
          "members": [
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.mbconv_params.MBConvParams.model_config",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "model_config",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "model_config = ConfigDict(frozen=True, extra='forbid')",
              "source": {
                "line": 21,
                "path": "helia_edge/layers/mbconv_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L21"
              },
              "summary": ""
            },
            {
              "description": "Number of filters",
              "examples": [],
              "id": "helia_edge.layers.mbconv_params.MBConvParams.filters",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "filters",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "filters: int = Field(..., description='# filters')",
              "source": {
                "line": 23,
                "path": "helia_edge/layers/mbconv_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L23"
              },
              "summary": ""
            },
            {
              "description": "Layer depth",
              "examples": [],
              "id": "helia_edge.layers.mbconv_params.MBConvParams.depth",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "depth",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "depth: int = Field(default=1, description='Layer depth')",
              "source": {
                "line": 24,
                "path": "helia_edge/layers/mbconv_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L24"
              },
              "summary": ""
            },
            {
              "description": "Expansion ratio",
              "examples": [],
              "id": "helia_edge.layers.mbconv_params.MBConvParams.ex_ratio",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "ex_ratio",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "ex_ratio: float = Field(default=1, description='Expansion ratio')",
              "source": {
                "line": 25,
                "path": "helia_edge/layers/mbconv_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L25"
              },
              "summary": ""
            },
            {
              "description": "Kernel size",
              "examples": [],
              "id": "helia_edge.layers.mbconv_params.MBConvParams.kernel_size",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "kernel_size",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "kernel_size: int | tuple[int, int] = Field(default=3, description='Kernel size')",
              "source": {
                "line": 26,
                "path": "helia_edge/layers/mbconv_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L26"
              },
              "summary": ""
            },
            {
              "description": "Stride size",
              "examples": [],
              "id": "helia_edge.layers.mbconv_params.MBConvParams.strides",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "strides",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "strides: int | tuple[int, int] = Field(default=1, description='Stride size')",
              "source": {
                "line": 27,
                "path": "helia_edge/layers/mbconv_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L27"
              },
              "summary": ""
            },
            {
              "description": "Squeeze Excite ratio",
              "examples": [],
              "id": "helia_edge.layers.mbconv_params.MBConvParams.se_ratio",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "se_ratio",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "se_ratio: float = Field(default=8, description='Squeeze Excite ratio')",
              "source": {
                "line": 28,
                "path": "helia_edge/layers/mbconv_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L28"
              },
              "summary": ""
            },
            {
              "description": "Drop rate",
              "examples": [],
              "id": "helia_edge.layers.mbconv_params.MBConvParams.droprate",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "droprate",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "droprate: float = Field(default=0, description='Drop rate')",
              "source": {
                "line": 29,
                "path": "helia_edge/layers/mbconv_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L29"
              },
              "summary": ""
            },
            {
              "description": "Batch normalization momentum",
              "examples": [],
              "id": "helia_edge.layers.mbconv_params.MBConvParams.bn_momentum",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "bn_momentum",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "bn_momentum: float = Field(default=0.9, description='Batch normalization momentum')",
              "source": {
                "line": 30,
                "path": "helia_edge/layers/mbconv_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L30"
              },
              "summary": ""
            },
            {
              "description": "Activation function",
              "examples": [],
              "id": "helia_edge.layers.mbconv_params.MBConvParams.activation",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "activation",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "activation: str = Field(default='relu6', description='Activation function')",
              "source": {
                "line": 31,
                "path": "helia_edge/layers/mbconv_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L31"
              },
              "summary": ""
            }
          ],
          "name": "MBConvParams",
          "params": [],
          "raises": [],
          "returns": [],
          "signature": "MBConvParams()",
          "source": {
            "line": 6,
            "path": "helia_edge/layers/mbconv_params.py",
            "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/mbconv_params.py#L6"
          },
          "summary": "MBConv parameters"
        }
      ]
    }
  ],
  "name": "helia_edge"
}