diff options
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) |