wb14123/seq2seq-couplet

self.eval_session.run()运算出来的结果维度有问题,会导致解码时代码报错

longglecc opened this issue · 3 comments

self.eval_session.run()运算出来的结果维度有问题,会导致解码时代码报错

Please provide error stack

是的,我也出现了这个问题。model.py里面train的时候没问题,但是bleu的eval中,outputs维度是三维,就算取第i个,也依然有2维,无法进行接下来的decode操作,请问这是什么原因?我看了下seq2seq代码里面train和eval 进行的output返回值就有差异,有没有人知道怎么解决啊

cui-z commented

因为eval使用了BeamSearchDecoder 所以输出是多个候选集,一般选择出第一条数据即可