convmixer_params

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

family: Literal['convmixer'] = 'convmixer'

Model family

filters: int = Field(default=256, description='# filters per layer')

Number of filters per layer

depth: int = Field(default=8, description='Network depth')

Network depth

kernel_size: int = Field(default=5, description='Filter size')

Filter size

patch_size: int = Field(default=2, description='Patch size')

Patch size

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

Include top

num_classes: int | None = Field(default=None, gt=0, description='Classes of the dense head; None for no dense layer')

Classes of the dense head; None for no dense layer

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

Output activation