# helia_edge.models.convmixer

## ConvMixer

For more info, refer to the original paper [ConvMixer: Revisiting Convolution in Vision](https://arxiv.org/abs/2201.09792).

**Functions**

| Name | Description |
| --- | --- |
| `build` | ConvMixer model from ``ConvMixerParams`` |
| `conv_mixer_block` | ConvMixer block |
| `conv_mixer_layer` | ConvMixer layer |

## helia_edge.models.convmixer.conv_mixer_block

`function` · `python`

```python
conv_mixer_block(filters: int, kernel_size: int) -> Callable[[keras.KerasTensor], keras.KerasTensor]
```

ConvMixer block

**Parameters**

| Name | Type | Default | Description |
| --- | --- | --- | --- |
| filters | int | Required | Number of filters |
| kernel_size | int | Required | Kernel size |

Source: `helia_edge/models/convmixer.py:21`

## helia_edge.models.convmixer.conv_mixer_layer

`function` · `python`

```python
conv_mixer_layer(x: keras.KerasTensor, params: ConvMixerParams) -> keras.KerasTensor
```

ConvMixer: https://openreview.net/pdf?id=TVHS5Y4dNvM.

**Parameters**

| Name | Type | Default | Description |
| --- | --- | --- | --- |
| x | keras.KerasTensor | Required | Input tensor |
| params | ConvMixerParams | Required | Model parameters. |

**Returns**

| Name | Type | Description |
| --- | --- | --- |
|  | keras.KerasTensor | keras.KerasTensor: Model output |

Source: `helia_edge/models/convmixer.py:47`

## helia_edge.models.convmixer.build

`function` · `python`

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

Build a ConvMixer model.

**Parameters**

| Name | Type | Default | Description |
| --- | --- | --- | --- |
| params | ConvMixerParams | Required | Model parameters. |
| input_shape | tuple[int \| None, ...] | Required | Input shape without the batch axis; None for a variable axis. |
| batch_size | int \| None | None | Static batch size; None for a dynamic batch. |
| name | str \| None | None | Model name; the family when None. |

**Returns**

| Name | Type | Description |
| --- | --- | --- |
|  | keras.Model | keras.Model: The model, named ``convmixer`` unless ``name`` is given. |

Source: `helia_edge/models/convmixer.py:83`