experiment: make ResNet 27-cell R2 executable

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,
     }