facebookresearch/ParlAI

[bert_ranker] Should be overriding build_criterion

Open

#2 019 ouverte le 26 sept. 2019

Voir sur GitHub
 (1 commentaire) (0 réactions) (0 assignés)Python (2 089 forks)batch import
CS5152Help WantedMinordonotreap

Métriques du dépôt

Stars
 (10 371 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

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

Guide contributeur