Function run_obj_ptr

tests/dump_phase7_reference.py:573–589

(
    tracker_weights: Dict[str, torch.Tensor],
    sam_token: torch.Tensor,
    obj_score_logit: torch.Tensor,
)

def run_obj_ptr(
    tracker_weights: Dict[str, torch.Tensor],
    sam_token: torch.Tensor,
    obj_score_logit: torch.Tensor,
) -> torch.Tensor:
    h = sam_token
    for i in range(3):
        h = F.linear(
            h,
            tracker_weights[f"obj_ptr_proj.layers.{i}.weight"].float(),
            tracker_weights[f"obj_ptr_proj.layers.{i}.bias"].float(),
        )
        if i < 2:
            h = F.relu(h)
    no_obj_ptr = tracker_weights["no_obj_ptr"].float()
    is_obj = (obj_score_logit > 0).float()
    return is_obj * h + (1.0 - is_obj) * no_obj_ptr


def build_prompt_and_pos(