facebookresearch/ParlAI

[bert_ranker] Should be overriding build_criterion

Open

#2,019 opened on 2019年9月26日

GitHub で見る
 (1 comment) (0 reactions) (0 assignees)Python (10,371 stars) (2,089 forks)batch import
CS5152Help WantedMinordonotreap

説明

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

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

[bert_ranker] Should be overriding build_criterion · facebookresearch/ParlAI#2019 | Good First Issue