metaformer_params

Typed parameters of the metaformer family; importable without Keras.

NameArgs()

Name and arguments

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

name: str = Field(default='conv', description='Name')

Name

args: dict = Field(default_factory=dict, description='Arguments')

Arguments

MetaFormerBlockParams()

MetaFormer block parameters

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

layers: int = Field(default=2, description='Number of layers')

Number of layers

patch_embed: dict = Field(default_factory=dict, description='Patch embedding')

Patch embedding

token_mixer: NameArgs = Field(default_factory=dict, description='Token mixer')

Token mixer

channel_mixer: NameArgs = Field(default_factory=dict, description='Channel mixer')

Channel mixer

MetaFormerParams()

MetaFormer parameters

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

family: Literal['metaformer'] = 'metaformer'

Model family

blocks: list[MetaFormerBlockParams] = Field(default_factory=list, description='MetaFormer blocks')

MetaFormer blocks

output_filters: int = Field(default=0, description='Output filters')

Output filters

output_activation: str | None = Field(default=None, description='Output activation')

Output activation

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

Include top

num_classes: int | None = Field(default=None, gt=0, description='Classes of the output layer')

Classes of the output layer

dropout: float = Field(default=0.2, description='Dropout rate')

Dropout rate

drop_connect_rate: float = Field(default=0.2, description='Drop connect rate')

Drop connect rate