Update evo_architecture.py

evo_architecture.py

@@ -37,8 +37,8 @@ def mutate_genome(base_config, exploration_rate=0.5):
         delta = random.randint(-change, change)
         config[mutation_type] = max(min_val, min(max_val, config[mutation_type] + delta))
 
-    # ✅ Fix: Ensure embed_dim is divisible by num_heads
-    embed_dim = 512  # Your hardcoded d_model
+    # ✅ Ensure num_heads divides d_model cleanly
+    embed_dim = 512
     if config["num_heads"] > embed_dim:
         config["num_heads"] = max(1, embed_dim // 64)
     while embed_dim % config["num_heads"] != 0:
@@ -52,10 +52,10 @@ def mutate_genome(base_config, exploration_rate=0.5):
 def log_genome(config, score=None):
     row = [
         config.get("genome_id", ""),
-        config["num_layers"],
-        config["ffn_dim"],
-        config["num_heads"],
-        config["memory_enabled"],
+        config.get("num_layers", ""),
+        config.get("ffn_dim", ""),
+        config.get("num_heads", ""),
+        config.get("memory_enabled", ""),
         score if score is not None else ""
     ]
     file_exists = os.path.exists(GENOME_LOG)