feat: add person fusion screening variants

infer_turn_head.py

@@ -14,6 +14,7 @@ from tqdm.auto import tqdm
 from easycom_head import (
     CLASS_NAMES,
     DATASET_NAMES,
+    LOCAL_FUSION_SCOPES,
     NUM_CLASSES,
     NUM_HORIZONS,
     OFFICIAL_TEST_SESSIONS,
@@ -418,6 +419,15 @@ def main() -> None:
     print(
         f"loss: {loss_name}; output_space: {output_space_for_loss(loss_name)}"
     )
+    print(
+        f"local fusion: {head.local_fusion}; "
+        f"scope: {LOCAL_FUSION_SCOPES[head.local_fusion]}"
+    )
+    if head.person_interaction_gates():
+        print(
+            "person interaction gates: "
+            + json.dumps(head.person_interaction_gates(), sort_keys=True)
+        )
     print(f"sessions: {', '.join(sessions)}")
     for name in sessions:
         data = load_session_data(
@@ -513,6 +523,8 @@ def main() -> None:
             "checkpoint_dev_sessions": checkpoint_payload.get("dev_sessions"),
             "layer_fusion": head.layer_fusion,
             "local_fusion": head.local_fusion,
+            "local_fusion_scope": LOCAL_FUSION_SCOPES[head.local_fusion],
+            "person_interaction_gates": head.person_interaction_gates(),
             "local_contract": local_contract,
             "transition_auxiliary": head.transition_auxiliary,
             "transition_objective": checkpoint_payload.get(