Add causal oracle person information screen
This commit is contained in:
@@ -63,6 +63,7 @@ LAYER_FUSION_MODES = (
|
||||
)
|
||||
CURRENT_FRAME_PERSON_SET_FUSION_MODES = (
|
||||
"frame-set-attention",
|
||||
"oracle-frame-set-attention",
|
||||
"audio-person-frame-set",
|
||||
"gated-person-state-adapter",
|
||||
)
|
||||
@@ -75,6 +76,7 @@ LOCAL_FUSION_MODES = ("pooled", *PERSON_SET_FUSION_MODES)
|
||||
LOCAL_FUSION_SCOPES = {
|
||||
"pooled": "pooled-baseline",
|
||||
"frame-set-attention": "f1-current-frame-person-set",
|
||||
"oracle-frame-set-attention": "causal-oracle-visible-person-speaking-history",
|
||||
"audio-person-frame-set": "f5-screening-current-frame-no-person-history",
|
||||
"gated-person-state-adapter": "f6-screening-post-thinker-not-internal",
|
||||
"track-gru-residual": "causal-anonymous-track-gru-pooled-baseline-residual",
|
||||
@@ -1303,6 +1305,7 @@ def load_session_data(
|
||||
sample_id=name,
|
||||
dataset=dataset,
|
||||
expected_contract=local_contract,
|
||||
oracle_analysis=local_fusion == "oracle-frame-set-attention",
|
||||
)
|
||||
else:
|
||||
from local_turn_features import load_local_features
|
||||
|
||||
Reference in New Issue
Block a user