"fix frigled test gradient of rnn" (#7166)

* "fix frigled test gradient of rnn" * "fix based on comments"
7 years ago · 19541468b6
parent 231e2ee55e
commit 19541468b6
1 changed files with 4 additions and 2 deletions
--- a/paddle/gserver/tests/test_LayerGrad.cpp
+++ b/paddle/gserver/tests/test_LayerGrad.cpp
@ -1472,7 +1472,8 @@ TEST(Layer, RecurrentLayer) {
    for (auto reversed : {false, true}) {
      config.layerConfig.set_reversed(reversed);
      config.testState = !reversed;
-      testLayerGrad(config, "recurrent", 50, /* trans= */ false, useGpu);
+      testLayerGrad(
+          config, "recurrent", 50, /* trans= */ false, useGpu, false, 1.0);
    }
  }
 }
@ -1494,7 +1495,8 @@ TEST(Layer, LstmLayer) {
    for (auto reversed : {false, true}) {
      config.layerConfig.set_reversed(reversed);
      config.testState = !reversed;
-      testLayerGrad(config, "lstmemory", 100, /* trans= */ false, useGpu);
+      testLayerGrad(
+          config, "lstmemory", 100, /* trans= */ false, useGpu, false, 0.02);
    }
  }
  for (auto useGpu : {true}) {