maybe bug in model_builder_rel.py(pytorch1_0 version) ?

[yrcong]

Hi,

for the code of pytorch1.0 version, in model_builder_rel.py line 504

_, prd_bias_scores_sbj_pos, _, ttl_cls_scores_sbj_pos, _, _, prd_cls_feats = \ self.RelDN(spo_feat_sbj_pos, spt_feat_sbj_pos, sbj_labels_sbj_pos_fg, obj_labels_sbj_pos_fg, sbj_feat_sbj_pos, obj_feat_sbj_pos)

7 elements are expected but RelDN (reldn_heads.py) only returns 6 (line 136).
return prd_vis_scores, prd_bias_scores, prd_spt_scores, ttl_cls_scores, sbj_cls_scores, obj_cls_scores

For the master branch there is not this problem..

Is this only a bug or something that i didn't understand or made mistakes?

thanks a lot