fix: model somato-dendritic baseline per neuron

1 files changed, 4 insertions, 2 deletions

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)