# helia_edge.models.unet_params

Typed parameters of the unet family; importable without Keras.

## helia_edge.models.unet_params.UNetBlockParams

`class` · `python`

```python
UNetBlockParams()
```

UNet block parameters

Source: `helia_edge/models/unet_params.py:8`

### helia_edge.models.unet_params.UNetBlockParams.model_config

`attribute` · `python`

```python
model_config = ConfigDict(frozen=True, extra='forbid')
```

Source: `helia_edge/models/unet_params.py:26`

### helia_edge.models.unet_params.UNetBlockParams.filters

`attribute` · `python`

```python
filters: int = Field(..., description='# filters')
```

Number of filters

Source: `helia_edge/models/unet_params.py:28`

### helia_edge.models.unet_params.UNetBlockParams.depth

`attribute` · `python`

```python
depth: int = Field(default=1, description='Layer depth')
```

Layer depth

Source: `helia_edge/models/unet_params.py:29`

### helia_edge.models.unet_params.UNetBlockParams.ddepth

`attribute` · `python`

```python
ddepth: int | None = Field(default=None, description='Decoder depth')
```

Decoder depth

Source: `helia_edge/models/unet_params.py:30`

### helia_edge.models.unet_params.UNetBlockParams.kernel

`attribute` · `python`

```python
kernel: int | tuple[int, int] = Field(default=3, description='Kernel size')
```

Kernel size

Source: `helia_edge/models/unet_params.py:31`

### helia_edge.models.unet_params.UNetBlockParams.pool

`attribute` · `python`

```python
pool: int | tuple[int, int] = Field(default=3, description='Pool size')
```

Pool size

Source: `helia_edge/models/unet_params.py:32`

### helia_edge.models.unet_params.UNetBlockParams.strides

`attribute` · `python`

```python
strides: int | tuple[int, int] = Field(default=1, description='Stride size')
```

Stride size

Source: `helia_edge/models/unet_params.py:33`

### helia_edge.models.unet_params.UNetBlockParams.skip

`attribute` · `python`

```python
skip: bool = Field(default=True, description='Add skip connection')
```

Add skip connection

Source: `helia_edge/models/unet_params.py:34`

### helia_edge.models.unet_params.UNetBlockParams.seperable

`attribute` · `python`

```python
seperable: bool = Field(default=False, description='Use seperable convs')
```

Use seperable convs

Source: `helia_edge/models/unet_params.py:35`

### helia_edge.models.unet_params.UNetBlockParams.dropout

`attribute` · `python`

```python
dropout: float | None = Field(default=None, description='Dropout rate')
```

Dropout rate

Source: `helia_edge/models/unet_params.py:36`

### helia_edge.models.unet_params.UNetBlockParams.norm

`attribute` · `python`

```python
norm: Literal['batch', 'layer'] | None = Field(default='batch', description='Normalization type')
```

Normalization type

Source: `helia_edge/models/unet_params.py:37`

### helia_edge.models.unet_params.UNetBlockParams.activation

`attribute` · `python`

```python
activation: Literal['relu', 'relu6', 'leaky_relu', 'elu', 'selu'] = Field(default='relu6', description='Activation')
```

Activation

Source: `helia_edge/models/unet_params.py:38`

### helia_edge.models.unet_params.UNetBlockParams.dilation

`attribute` · `python`

```python
dilation: int | tuple[int, int] | None = Field(default=None, description='Dilation factor')
```

Dilation factor

Source: `helia_edge/models/unet_params.py:39`

## helia_edge.models.unet_params.UNetParams

`class` · `python`

```python
UNetParams()
```

UNet parameters

Source: `helia_edge/models/unet_params.py:42`

### helia_edge.models.unet_params.UNetParams.model_config

`attribute` · `python`

```python
model_config = ConfigDict(frozen=True, extra='forbid')
```

Source: `helia_edge/models/unet_params.py:56`

### helia_edge.models.unet_params.UNetParams.family

`attribute` · `python`

```python
family: Literal['unet'] = 'unet'
```

Model family

Source: `helia_edge/models/unet_params.py:58`

### helia_edge.models.unet_params.UNetParams.blocks

`attribute` · `python`

```python
blocks: list[UNetBlockParams] = Field(default_factory=list, description='UNet blocks')
```

UNet blocks

Source: `helia_edge/models/unet_params.py:59`

### helia_edge.models.unet_params.UNetParams.include_top

`attribute` · `python`

```python
include_top: bool = Field(default=True, description='Include top')
```

Include top

Source: `helia_edge/models/unet_params.py:60`

### helia_edge.models.unet_params.UNetParams.num_classes

`attribute` · `python`

```python
num_classes: int | None = Field(default=None, gt=0, description='Classes of the output layer; required with include_top')
```

Classes of the output layer; required with include_top

Source: `helia_edge/models/unet_params.py:61`

### helia_edge.models.unet_params.UNetParams.use_logits

`attribute` · `python`

```python
use_logits: bool = Field(default=True, description='Use logits')
```

Use logits

Source: `helia_edge/models/unet_params.py:64`

### helia_edge.models.unet_params.UNetParams.output_kernel_size

`attribute` · `python`

```python
output_kernel_size: int | tuple[int, int] = Field(default=3, description='Output kernel size')
```

Output kernel size

Source: `helia_edge/models/unet_params.py:65`

### helia_edge.models.unet_params.UNetParams.output_kernel_stride

`attribute` · `python`

```python
output_kernel_stride: int | tuple[int, int] = Field(default=1, description='Output kernel stride')
```

Output kernel stride

Source: `helia_edge/models/unet_params.py:66`