Update config.py

This commit is contained in:
Zheyuan Wu
2026-03-16 15:12:32 -05:00
parent 18dd58c0dd
commit e58114cb3f

View File

@@ -19,8 +19,9 @@ RESULTS_DIR = (
# 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)]
# SPHERE_DIMS = [1<<i for i in range(4, 12)]
SPHERE_DIMS = []
CP_DIMS = [(1<<i, 1<<i) for i in range(9, 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.