提交 a4f5cc03 编写于 作者: S sneaxiy

test=release/1.0.0

上级 f158829d
fsdfgdfsdsafsdg
...@@ -88,7 +88,7 @@ class TestBase(unittest.TestCase): ...@@ -88,7 +88,7 @@ class TestBase(unittest.TestCase):
self.epsilon = epsilon self.epsilon = epsilon
self.decay = 0.9 self.decay = 0.9
self.momentum = 0.0 self.momentum = 0.1
self.centered = centered self.centered = centered
self.ms_out = self.decay * self.mean_square + (1 - self.decay self.ms_out = self.decay * self.mean_square + (1 - self.decay
...@@ -170,8 +170,11 @@ class TestRmspropOp(TestBase): ...@@ -170,8 +170,11 @@ class TestRmspropOp(TestBase):
rmsprop_op.run(self.scope, self.place) rmsprop_op.run(self.scope, self.place)
self.check( self.check(
np.array(self.mean_square_tensor), self.ms_out, self.place, np.array(self.mean_square_tensor),
self.mean_square_name) self.ms_out,
self.place,
self.mean_square_name,
atol=atol)
self.check( self.check(
np.array(self.moment_tensor), np.array(self.moment_tensor),
self.moment_out, self.moment_out,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册