Machine-Learning
/
AIIA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

kernel size as large as channel size

This commit is contained in:
Falko Victor Habel 2025-01-26 23:26:42 +01:00
parent b501ae8317
commit 29f0d86ff7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/aiia/model/config.py
View File

@ -8,7 +8,7 @@ class AIIAConfig:
def __init__( def __init__(
self, self,
model_name: str = "AIIA", model_name: str = "AIIA",
kernel_size: int = 5, kernel_size: int = 3,
activation_function: str = "GELU", activation_function: str = "GELU",
hidden_size: int = 512, hidden_size: int = 512,
num_hidden_layers: int = 12, num_hidden_layers: int = 12,