corrected activation function

2025-01-22 11:20:05 +01:00 · 2025-01-22 11:20:05 +01:00 · 26b701fd77
parent 74973a325b
commit 26b701fd77
1 changed files with 1 additions and 1 deletions
--- a/src/aiia/model/config.py
+++ b/src/aiia/model/config.py
@ -7,7 +7,7 @@ class AIIAConfig:
        self,
        model_name: str = "AIIA",
        kernel_size: int = 3,
-        activation_function: str = "gelu",
+        activation_function: str = "GELU",
        hidden_size: int = 128,
        num_hidden_layers: int = 2,
        num_channels: int = 3,