[TRA] Add option to return scores (#2427)
[TRA] Add option to return scores (#2427)

* return scores from tra

* lintitlkasj

* renmae

facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI