facebookresearch/ParlAI

[bert_ranker] Should be overriding build_criterion

Aberta

#2.019 aberto em 26 de set. de 2019

 (1 comentário) (0 reação) (0 responsável)Python (2.089 forks)batch import
CS5152Help WantedMinordonotreap

Métricas do repositório

Stars
 (10.371 estrelas)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 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

Guia do colaborador