From 5255166d13c46402deb2b43cfeafd8953c787fec Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Tue, 21 Jul 2026 06:46:33 -0500 Subject: fix: model somato-dendritic baseline per neuron --- experiments/run.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'experiments/run.py') diff --git a/experiments/run.py b/experiments/run.py index 3ad0bdd..bb2d151 100644 --- a/experiments/run.py +++ b/experiments/run.py @@ -45,13 +45,14 @@ def build(args, device): sizes = [args.n_in] + [args.width] * args.depth + [10] if args.mode == "bp": net = BPNet(sizes, act=args.act, device=device, seed=args.seed, - w_scale=args.w_scale, nuis_rho=0.0, residual=bool(args.residual)) + w_scale=args.w_scale, nuis_rho=0.0, residual=bool(args.residual), + predictor_mode=args.predictor_mode) cfg = SDILConfig(eta=args.eta, momentum=args.momentum) return net, cfg net = SDILNet(sizes, act=args.act, device=device, seed=args.seed, w_scale=args.w_scale, a_scale=args.a_scale, nuis_rho=args.nuis_rho, feedback=args.feedback, - residual=bool(args.residual)) + residual=bool(args.residual), predictor_mode=args.predictor_mode) if args.mode == "dfa": cfg = dfa_config(eta=args.eta, momentum=args.momentum) elif args.mode == "sdil": @@ -175,6 +176,7 @@ def get_args(): p.add_argument("--p_warmup_steps", type=int, default=200) # pre-task neutral P warmup p.add_argument("--p_warmup_eta", type=float, default=0.05) p.add_argument("--nuis_rho", type=float, default=0.0) + p.add_argument("--predictor_mode", default="diagonal", choices=["diagonal", "full"]) p.add_argument("--normalize_delta", type=int, default=0) p.add_argument("--settle_steps", type=int, default=0) p.add_argument("--kappa", type=float, default=0.0) -- cgit v1.2.3