Upload ai/overnight_evolution.py with huggingface_hub

ai/overnight_evolution.py

@@ -0,0 +1,58 @@
+import os
+import subprocess
+import time
+
+
+def run_step(name, cmd):
+    print(f"\n>>> Starting {name}...")
+    start = time.time()
+    try:
+        subprocess.check_call(cmd, shell=True)
+        print(f">>> {name} completed in {time.time() - start:.1f}s")
+    except subprocess.CalledProcessError as e:
+        print(f"!!! Error in {name}: {e}")
+        return False
+    return True
+
+
+def main():
+    print("=== LOVE-CA ALPHA ZERO OVERNIGHT EVOLUTION ===")
+
+    # 1. Ensure model is exported
+    if not os.path.exists("ai/models/alphanet.onnx"):
+        run_step("Model Export", "uv run python ai/utils/export_onnx.py")
+
+    generation = 0
+    while True:
+        print(f"\n--- GENERATION {generation} ---")
+
+        # 1. Generate Data (Self-Play)
+        # Using 800 sims for high quality, 1000 games per batch
+        if not run_step(
+            "Data Generation",
+            "uv run python ai/data_generation/generate_data.py --num-games 1000 --sims 800 --output-file ai/data/overnight_batch.npz",
+        ):
+            break
+
+        # 2. Train Model
+        # Training for a few epochs on the new data
+        if not run_step(
+            "Model Training", "uv run python ai/training/train.py --data-path ai/data/overnight_batch.npz --epochs 5"
+        ):
+            break
+
+        # 3. Export New Model
+        run_step("Model Update", "uv run python ai/utils/export_onnx.py")
+
+        # 4. Evaluation (Optional, simple score)
+        print(f"Generation {generation} completed.")
+
+        generation += 1
+
+        # Limit or break? User said overnight, so loop forever until stopped.
+        # But maybe sleep a bit to avoid CPU melt
+        time.sleep(10)
+
+
+if __name__ == "__main__":
+    main()