未验证 提交 19541468 编写于 作者: D dzhwinter 提交者: GitHub

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

* "fix frigled test gradient of rnn"

* "fix based on comments"
上级 231e2ee5
...@@ -1472,7 +1472,8 @@ TEST(Layer, RecurrentLayer) { ...@@ -1472,7 +1472,8 @@ TEST(Layer, RecurrentLayer) {
for (auto reversed : {false, true}) { for (auto reversed : {false, true}) {
config.layerConfig.set_reversed(reversed); config.layerConfig.set_reversed(reversed);
config.testState = !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) { ...@@ -1494,7 +1495,8 @@ TEST(Layer, LstmLayer) {
for (auto reversed : {false, true}) { for (auto reversed : {false, true}) {
config.layerConfig.set_reversed(reversed); config.layerConfig.set_reversed(reversed);
config.testState = !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}) { for (auto useGpu : {true}) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册