facebookresearch/ParlAI

[bert_ranker] Should be overriding build_criterion

Open

#2.019 aberto em 26 de set. de 2019

Ver no GitHub
 (1 comment) (0 reactions) (0 assignees)Python (2.089 forks)batch import
CS5152Help WantedMinordonotreap

Métricas do repositório

Stars
 (10.371 stars)
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