diff --git a/src/aiia/model/config.py b/src/aiia/model/config.py index e2ae83e..02bc709 100644 --- a/src/aiia/model/config.py +++ b/src/aiia/model/config.py @@ -8,7 +8,7 @@ class AIIAConfig: def __init__( self, model_name: str = "AIIA", - kernel_size: int = 5, + kernel_size: int = 3, activation_function: str = "GELU", hidden_size: int = 512, num_hidden_layers: int = 12,