Commit f4be1882 by xiaotong

correct output length for dumping

parent 12195a67
......@@ -365,8 +365,8 @@ void T2TTrainer::Test(const char * fn, const char * ofn, T2TModel * model)
ShowNTErrors("Illegal model type!");
}
int bSize = batchDec.GetDim(0);
int length = batchDec.GetDim(1);
int bSize = output.GetDim(0);
int length = output.GetDim(1);
/* prediction probabilities */
XTensor probs;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论