From 29f0d86ff7c9360437ae36500142115174e034af Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Sun, 26 Jan 2025 23:26:42 +0100 Subject: [PATCH] kernel size as large as channel size --- src/aiia/model/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,