Skip to content

ctde_v0.train_ctde

train_ctde

Entry point: train the grounded CTDE v0 agent (LPAC + goal-pointer + λ̂₂).

Short CPU validation (16x16 / 4 agents): JAX_PLATFORMS=cpu /Users/bijanmehr/Project.Zymera/zymera_lab/.venv/bin/python -m ctde_v0.train_ctde --iters 20 --rollouts 8

Longer run (with a checkpoint + saved config sidecar): JAX_PLATFORMS=cpu /Users/bijanmehr/Project.Zymera/zymera_lab/.venv/bin/python -m ctde_v0.train_ctde --iters 200 --rollouts 16 --grid 16 --n-agents 4 --agg max --mechanism action_mask --run-dir runs/ctde_v0_16x16x4

Per iter it logs: episode reward, coverage %, connectivity %, mean λ₂, and the aux-λ₂ accuracy (1 - median rel-err of λ̂₂ vs true λ₂), plus the controller valid-move fraction (should be 100%). --run-dir writes config.json, history.json and (if asked) a checkpoint next to the logs.