facebookresearch/ParlAI

[bert_ranker] Should be overriding build_criterion

オープン

#2,019 opened on 2019/09/26

 (1 件のコメント) (0 件のリアクション) (0 人の担当者)Python (2,089 件のフォーク)batch import
CS5152Help WantedMinordonotreap

Repository metrics

Stars
 (10,371 個のスター)
PR merge metrics
 (PR metrics pending)

説明

Bug description Created from #2003.

The self.rank_loss in bi_encode_ranker (and possibly the others) should be created by overriding the build_criterion rather than directionly setting self.rank_loss.

https://github.com/facebookresearch/ParlAI/blob/028753ff0cbc293037117f1b0bb07eb7e7335093/parlai/agents/bert_ranker/bi_encoder_ranker.py#L57

コントリビューターガイド