updates?

codes/experiment_v0.2/config.py

@@ -1,24 +1,27 @@
-"""Edit globals here; no CLI parser is used."""
-from datetime import datetime
-from pathlib import Path
-
-SEED = 7
-KAPPA = 1e-3
-NUM_SAMPLES = 10**4                  # requested default
-LIPSCHITZ_PAIRS = 12_000
-LIPSCHITZ_RESERVOIR = 4_096
-MAJORANA_STAR_STATES = 16            # only for visualization
-MAX_STAR_DEGREE = 63                 # avoid unstable huge root-finding plots
-
-BACKEND = "auto"                     # auto | jax | numpy
-JAX_PLATFORM = ""                    # "", "cpu", "gpu"; set before importing JAX
-RESULTS_DIR = Path("./results") / f"exp-{datetime.now():%Y%m%d-%H%M%S}"
-
-# Chosen so the three families have comparable intrinsic dimensions:
-# sphere S^(m-1), CP^(d_A d_B - 1), and Sym^N(C^2) ~ CP^N.
-SPHERE_DIMS = [16, 64, 256, 1024]
-CP_DIMS = [(4, 4), (8, 8), (16, 16), (32, 32)]
-MAJORANA_N = [15, 63, 255, 1023]
-
-# Batch sizes are the main speed knob; reduce CP batches first if memory is tight.
-BATCH = {"sphere": 32_768, "cp": 256, "majorana": 65_536}
+"""Edit globals here; no CLI parser is used."""
+
+from datetime import datetime
+from pathlib import Path
+
+SEED = 114514
+KAPPA = 1e-3
+NUM_SAMPLES = 10**6  # requested default
+LIPSCHITZ_PAIRS = 12_000
+LIPSCHITZ_RESERVOIR = 4_096
+MAJORANA_STAR_STATES = 16  # only for visualization
+MAX_STAR_DEGREE = 63  # avoid unstable huge root-finding plots
+
+BACKEND = "auto"  # auto | jax | numpy
+JAX_PLATFORM = "gpu"  # "", "cpu", "gpu"; set before importing JAX
+RESULTS_DIR = (
+    Path.joinpath(Path.cwd(), Path("./results")) / f"exp-{datetime.now():%Y%m%d-%H%M%S}"
+)
+
+# Chosen so the three families have comparable intrinsic dimensions:
+# sphere S^(m-1), CP^(d_A d_B - 1), and Sym^N(C^2) ~ CP^N.
+SPHERE_DIMS = [1<<i for i in range(4, 12)]
+CP_DIMS = [(1<<i, 1<<i) for i in range(4, 12)]
+MAJORANA_N = [(1<<i)-1 for i in range(4, 12)]
+
+# Batch sizes are the main speed knob; reduce CP batches first if memory is tight.
+BATCH = {"sphere": 32_768, "cp": 256, "majorana": 65_536}