sdil: normalize causal vectorizer regression

1 files changed, 5 insertions, 1 deletions

diff --git a/experiments/run.py b/experiments/run.py
index ced7585..0019c67 100644
--- a/experiments/run.py
+++ b/experiments/run.py
@@ -223,7 +223,9 @@ def build(args, device):
             kappa=args.kappa, feedback=args.feedback,
             p_update_on_neutral=bool(args.p_neutral),
             normalize_delta=bool(args.normalize_delta),
-            raw_scale_control=args.raw_scale_control)
+            raw_scale_control=args.raw_scale_control,
+            vectorizer_optimizer=args.vectorizer_optimizer,
+            vectorizer_eps=args.vectorizer_eps)
     elif args.mode == "nodepert":
         if args.pert_every != 1:
             raise ValueError("direct node perturbation requires --pert_every 1")
@@ -679,6 +681,8 @@ def get_args():
     p.add_argument("--predictor_mode", default="diagonal", choices=["diagonal", "full"])
     p.add_argument("--vectorizer_mode", default="linear",
                    choices=["linear", "soma_gated", "context_gated"])
+    p.add_argument("--vectorizer_optimizer", default="sgd", choices=["sgd", "nlms"])
+    p.add_argument("--vectorizer_eps", type=float, default=1e-6)
     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)