algorithm: calibrate feedback in representable subspace

1 files changed, 5 insertions, 0 deletions

diff --git a/experiments/conv_run.py b/experiments/conv_run.py
index 98c1f19..cb99545 100644
--- a/experiments/conv_run.py
+++ b/experiments/conv_run.py
@@ -111,6 +111,7 @@ def build(args):
         use_residual=bool(args.use_residual), nuisance_scale=args.nuisance_scale,
         pert_sigma=args.pert_sigma, pert_every=args.pert_every,
         pert_directions=args.pert_directions,
+        apical_calibration_mode=args.apical_calibration_mode,
         direct_node_perturbation=args.mode == "nodepert")
     config.validate()
     return net, config
@@ -481,6 +482,10 @@ def parse_args():
     parser.add_argument("--pert_sigma", type=float, default=0.01)
     parser.add_argument("--pert_every", type=int, default=4)
     parser.add_argument("--pert_directions", type=int, default=1)
+    parser.add_argument(
+        "--apical_calibration_mode",
+        choices=("unit_targets", "channel_subspace"),
+        default="unit_targets")
     parser.add_argument("--predictor_warmup_steps", type=int, default=0)
     parser.add_argument("--a_warmup_steps", type=int, default=0)
     parser.add_argument("--alignment_probe", type=int, default=0)