updated config t add kwrags to support future changes to the config for different models

src/aiia/model/config.py

@@ -11,7 +11,8 @@ class AIIAConfig:
         hidden_size: int = 128,
         num_hidden_layers: int = 2,
         num_channels: int = 3,
-        learning_rate: float = 5e5
+        learning_rate: float = 5e-5,
+        **kwargs
     ):
         self.model_name = model_name
         self.kernel_size = kernel_size
@@ -21,6 +22,10 @@ class AIIAConfig:
         self.num_channels = num_channels
         self.learning_rate = learning_rate
         
+        # Store additional keyword arguments as attributes
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
     @property
     def activation_function(self):
         return self._activation_function
@@ -35,7 +40,7 @@ class AIIAConfig:
 
     def save(self, file_path):
         with open(file_path, 'w') as f:
-            json.dump(self.__dict__, f)
+            json.dump(vars(self), f)
 
     @classmethod
     def load(cls, file_path):