facebookresearch/ParlAI

[bert_ranker] Should be overriding build_criterion

開放

#2,019 建立於 2019年9月26日

 (1 則留言) (0 個反應) (0 位負責人)Python (2,089 個分叉)batch import
CS5152Help WantedMinordonotreap

倉庫指標

星標
 (10,371 顆星)
PR 合併指標
 (30 天內沒有已合併 PR)

描述

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

貢獻者指南