facebookresearch/ParlAI

[bert_ranker] Should be overriding build_criterion

Open

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

在 GitHub 查看
 (1 评论) (0 反应) (0 负责人)Python (10,371 star) (2,089 fork)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

贡献者指南