diff options
Diffstat (limited to 'experiments/run.py')
| -rw-r--r-- | experiments/run.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/experiments/run.py b/experiments/run.py index 93c1862..3b0f961 100644 --- a/experiments/run.py +++ b/experiments/run.py @@ -190,7 +190,8 @@ def build(args, device): 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), - predictor_mode=args.predictor_mode) + predictor_mode=args.predictor_mode, + vectorizer_mode=args.vectorizer_mode) cfg = SDILConfig(eta=args.eta, momentum=args.momentum) return net, cfg if args.mode == "fa": @@ -198,12 +199,14 @@ def build(args, device): w_scale=args.w_scale, nuis_rho=0.0, residual=bool(args.residual), predictor_mode=args.predictor_mode, + vectorizer_mode=args.vectorizer_mode, b_scale=args.feedback_scale) return net, SDILConfig(eta=args.eta, momentum=args.momentum) 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), predictor_mode=args.predictor_mode, + vectorizer_mode=args.vectorizer_mode, traffic_mode=args.traffic_mode, nuis_seed=args.traffic_seed) if args.mode == "dfa": cfg = dfa_config(eta=args.eta, momentum=args.momentum) @@ -583,6 +586,8 @@ def get_args(): p.add_argument("--traffic_mode", default="soma", choices=["none", "soma", "topdown", "mixed"]) 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("--normalize_delta", type=int, default=0) p.add_argument("--settle_steps", type=int, default=0) p.add_argument("--kappa", type=float, default=0.0) |