facebookresearch/ParlAI

[bert_ranker] Should be overriding build_criterion

Open

#2,019 opened on Sep 26, 2019

View on GitHub
 (1 comment) (0 reactions) (0 assignees)Python (2,089 forks)batch import
CS5152Help WantedMinordonotreap

Repository metrics

Stars
 (10,371 stars)
PR merge metrics
 (No merged PRs in 30d)

Description

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

Contributor guide