diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-07-21 07:02:04 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-07-21 07:02:04 -0500 |
| commit | 0f107d1ee4e750421a0e95f8bd06bf0f629d2fa7 (patch) | |
| tree | 4b31652461d4befd58aafeef17758d4c198d7eb7 /experiments/run.py | |
| parent | c2769e6b18e299b2267662b7c48364e8c220f631 (diff) | |
feat: multiplex causal probes across hidden layers
Diffstat (limited to 'experiments/run.py')
| -rw-r--r-- | experiments/run.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/experiments/run.py b/experiments/run.py index bb2d151..eb9f356 100644 --- a/experiments/run.py +++ b/experiments/run.py @@ -61,6 +61,7 @@ def build(args, device): use_residual=bool(args.use_residual), learn_A=bool(args.learn_A), learn_P=bool(args.learn_P), pert_sigma=args.pert_sigma, pert_every=args.pert_every, pert_ndirs=args.pert_ndirs, + pert_mode=args.pert_mode, momentum=args.momentum, settle_steps=args.settle_steps, kappa=args.kappa, feedback=args.feedback, p_update_on_neutral=bool(args.p_neutral), @@ -169,6 +170,7 @@ def get_args(): p.add_argument("--pert_sigma", type=float, default=1e-2) p.add_argument("--pert_every", type=int, default=4) p.add_argument("--pert_ndirs", type=int, default=4) + p.add_argument("--pert_mode", default="layerwise", choices=["layerwise", "simultaneous"]) p.add_argument("--use_residual", type=int, default=1) p.add_argument("--learn_A", type=int, default=1) p.add_argument("--learn_P", type=int, default=1) |