facebookresearch/ParlAI

[bert_ranker] Should be overriding build_criterion

Open

#2019 aperta il 26 set 2019

Vedi su GitHub
 (1 commento) (0 reazioni) (0 assegnatari)Python (2089 fork)batch import
CS5152Help WantedMinordonotreap

Metriche repository

Star
 (10.371 star)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

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

Guida contributor