kernel size as large as channel size

2025-01-26 23:26:42 +01:00 · 2025-01-26 23:26:42 +01:00 · 29f0d86ff7
parent b501ae8317
commit 29f0d86ff7
1 changed files with 1 additions and 1 deletions
--- 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,