fix(reldn-heads): prd_bias_scores undefined when (+1 more)

[andrewwhitecdw]

This PR addresses 2 issues in lib/modeling_rel/reldn_heads.py.

Fix 1

fix: prd_bias_scores undefined when USE_FREQ_BIAS is False

Fix: Replace:

        sbj_cls_scores = None
        obj_cls_scores = None
            
        if cfg.MODEL.USE_FREQ_BIAS:

with:

        sbj_cls_scores = None
        obj_cls_scores = None
        prd_bias_scores = None
            
        if cfg.MODEL.USE_FREQ_BIAS:

Fix 2

fix: torch.tensor breaks graph/device for total scores

Fix: Replace:

            ttl_cls_scores = torch.tensor(prd_vis_scores)

with:

            ttl_cls_scores = prd_vis_scores.clone()

Files changed

• lib/modeling_rel/reldn_heads.py