diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-07-22 12:52:05 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-07-22 12:52:05 -0500 |
| commit | d841aa67aea64ef735c29e063829f36dc3c5d7de (patch) | |
| tree | 351a622eab30bf8fcd910f678ddc49961a8805de /experiments/conv_run.py | |
| parent | da109afa23146988188c3af910c6a6205f0c63d9 (diff) | |
algorithm: estimate causal credit in vectorizer space
Diffstat (limited to 'experiments/conv_run.py')
| -rw-r--r-- | experiments/conv_run.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/experiments/conv_run.py b/experiments/conv_run.py index a33005d..3c0de00 100644 --- a/experiments/conv_run.py +++ b/experiments/conv_run.py @@ -213,10 +213,11 @@ def run(args): "schema_version": 1, "protocol_family": "oral_a_cifar_local_resnet_development", "calibration_metric_space": ( - None if config is None else - ("channel_basis_moments" - if config.apical_calibration_mode == "channel_subspace" - else "full_hidden_field")), + None if config is None else { + "unit_targets": "full_hidden_field", + "channel_subspace": "channel_basis_moments", + "vectorizer_subspace": "vectorizer_parameter_gradients", + }[config.apical_calibration_mode]), "args": vars(args), "provenance": provenance(), "split": split, @@ -496,7 +497,7 @@ def parse_args(): parser.add_argument("--pert_directions", type=int, default=1) parser.add_argument( "--apical_calibration_mode", - choices=("unit_targets", "channel_subspace"), + choices=("unit_targets", "channel_subspace", "vectorizer_subspace"), default="unit_targets") parser.add_argument("--predictor_warmup_steps", type=int, default=0) parser.add_argument("--a_warmup_steps", type=int, default=0) |