whr94621 / NJUNMT-pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

关于ensemble问题

Fengxiaoxx opened this issue · comments

你好,请问如果使用ensenmble解码的时候,默认的方法是什么啊,可不可以自己指定是使用平均还是投票呢

不好意思,工作后没时间在这个项目上了,导致刚看见这个issue🙇。我这里的实现是采取的概率平均方法进行ensemble的,实现的位置在

next_scores = torch.log(sum(next_probs_group_) / len(next_probs_group_))

如果是投票的话需要自己修改一下了