facebookresearch/ParlAI

[bert_ranker] Should be overriding build_criterion

Open

#2,019 创建于 2019年9月26日

在 GitHub 查看
 (1 评论) (0 反应) (0 负责人)Python (2,089 fork)batch import
CS5152Help WantedMinordonotreap

仓库指标

Star
 (10,371 star)
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

贡献者指南