diff options
Diffstat (limited to 'experiments/run.py')
| -rw-r--r-- | experiments/run.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/experiments/run.py b/experiments/run.py index aa509e4..a44e72e 100644 --- a/experiments/run.py +++ b/experiments/run.py @@ -101,7 +101,7 @@ def build(args, device): 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, - traffic_mode=args.traffic_mode) + traffic_mode=args.traffic_mode, nuis_seed=args.traffic_seed) if args.mode == "dfa": cfg = dfa_config(eta=args.eta, momentum=args.momentum) elif args.mode == "sdil": @@ -441,6 +441,8 @@ 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("--traffic_seed", type=int, default=1234, + help="ordinary apical-traffic projection seed; independent of model/data seeds") p.add_argument("--traffic_mode", default="soma", choices=["none", "soma", "topdown", "mixed"]) p.add_argument("--predictor_mode", default="diagonal", choices=["diagonal", "full"]) |