Update config.json

config.json

@@ -1,60 +1,52 @@
 {
-  "data_dir": "/path/to/data",
+  "model_type": "csumlm",
   "text_encoder": {
-    "model_path": "/path/to/text_encoder.h5",
-    "max_length": 512
+    "model_name": "bert-base-uncased",
+    "pretrained_weights": "path/to/bert-base-uncased"
   },
   "image_encoder": {
-    "model_path": "/path/to/image_encoder.h5",
-    "input_size": [224, 224, 3]
+    "model_name": "resnet50",
+    "pretrained_weights": "path/to/resnet50"
   },
   "audio_encoder": {
-    "model_path": "/path/to/audio_encoder.h5",
-    "sample_rate": 16000,
-    "duration": 5.0
+    "model_name": "wav2vec2-base",
+    "pretrained_weights": "path/to/wav2vec2-base"
   },
-  "hybrid_model": {
-    "dense_units": [512, 256],
-    "dropout_rate": 0.2,
-    "output_units": 1,
-    "output_activation": "sigmoid"
+  "fusion_layer": {
+    "type": "concatenate",
+    "output_dim": 1024
   },
-  "attention": {
+  "output_layer": {
+    "type": "dense",
+    "units": 1,
+    "activation": "sigmoid"
+  },
+  "attention_mechanism": {
     "traditional_attention": {
-      "attention_type": "additive"
+      "type": "bahdanau"
     },
     "self_attention": {
-      "num_heads": 8,
-      "key_dim": 64
+      "type": "scaled_dot_product"
     },
     "linear_attention": {
-      "num_heads": 4,
-      "key_dim": 32
+      "type": "linear"
     }
   },
   "belief_desire_intent_tree": {
-    "max_depth": 5,
-    "node_units": 128
+    "type": "recursive_neural_network"
   },
-  "training": {
-    "epochs": 10,
-    "batch_size": 32,
-    "learning_rate": 0.001
+  "iragel": {
+    "retrieval_model": "path/to/retrieval_model",
+    "generation_model": "path/to/generation_model"
+  },
+  "explainer": {
+    "type": "attention_visualization"
   },
   "realtime_learning": {
-    "update_interval": 100,
-    "update_batch_size": 16
+    "type": "online_learning",
+    "learning_rate": 0.001
   },
   "knowledge_base": {
-    "embedding_dim": 256,
-    "max_entities": 100000
-  },
-  "explainer": {
-    "explanation_type": "attention"
-  },
-  "iragel": {
-    "generation_length": 100,
-    "top_k": 5,
-    "top_p": 0.9
+    "type": "knowledge_graph_embeddings"
   }
 }