diff options
Diffstat (limited to 'experiments/resnet_crossover_native.py')
| -rw-r--r-- | experiments/resnet_crossover_native.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/experiments/resnet_crossover_native.py b/experiments/resnet_crossover_native.py index e346678..0c2ba97 100644 --- a/experiments/resnet_crossover_native.py +++ b/experiments/resnet_crossover_native.py @@ -196,6 +196,13 @@ def work_report(net, args, ordinary_examples, validation_examples, if args.method == "ep": relaxation_examples += ( args.ep_free_steps * validation_examples) + local_vjp_examples = ( + relaxation_examples + if args.method in ("ep", "dualprop") else 0 + ) + local_target_backward_examples = ( + 2 * ordinary_examples if args.method == "ff" else 0 + ) return { "forward_parameter_count": net.n_forward_parameters, "forward_macs_per_example": net.forward_macs_per_example, @@ -206,6 +213,10 @@ def work_report(net, args, ordinary_examples, validation_examples, "feedforward_example_passes": feedforward_examples, "relaxation_example_passes": relaxation_examples, "candidate_label_evaluation_presentations": candidate_examples, + "logical_task_loss_queries": 0, + "local_vjp_example_evaluations": local_vjp_examples, + "local_target_backward_example_evaluations": + local_target_backward_examples, "completed_global_epochs": completed_epochs, } |