update GradientMachine API

8 years ago · 258e5ec59f
parent 7c5fd23106
commit 258e5ec59f
1 changed files with 1 additions and 1 deletions
--- a/paddle/api/GradientMachine.cpp
+++ b/paddle/api/GradientMachine.cpp
@ -146,7 +146,7 @@ void GradientMachine::randParameters() { m->machine->randParameters(); }

 Matrix* GradientMachine::getLayerOutput(const std::string& layerName) const
    throw(UnsupportError) {
-  auto nn = std::dynamic_pointer_cast<paddle::NeuralNetwork>(m->machine);
+  auto nn = m->machine;
  if (nn) {
    auto mat = nn->getLayerOutput(layerName);
    return Matrix::createByPaddleMatrixPtr(&mat);