提交 ab8bb973 编写于 作者: Y Yu Yang 提交者: GitHub

Merge pull request #1373 from reyoung/feature/remove_grad_printer_seq_info

Remove sequence information in GradientPrinter.
Fix #1376
2 合并请求!11636[IMPORTANT] MKLDNN layout: Support for sum operator,!2081Release/0.10.0
......@@ -906,8 +906,6 @@ REGISTER_EVALUATOR(value_printer, ValuePrinter);
*/
class GradientPrinter : public Evaluator {
public:
GradientPrinter() {}
virtual void eval(const NeuralNetwork& nn) {
for (const std::string& name : config_.input_layers()) {
const Argument& argu = nn.getLayer(name)->getOutput();
......@@ -916,11 +914,6 @@ public:
argu.grad->print(os);
LOG(INFO) << "layer=" << name << " grad matrix:\n" << os.str();
}
if (auto startPos = argu.sequenceStartPositions) {
std::ostringstream os;
startPos->getVector(false)->print(os, startPos->getSize());
LOG(INFO) << "layer=" << name << " sequence pos vector:\n" << os.str();
}
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册
反馈
建议
客服 返回
顶部