提交 b0cb1c2b authored 作者: Frederic's avatar Frederic

Fix random test failure. Now it pass with seed 1 to 100

上级 5f42dd99
......@@ -212,8 +212,10 @@ class BlockSparse_Gemv_and_Outer(unittest.TestCase):
def op(b, h, W):
return self.gemv_op(b.take(oIdx, axis=0), W, h, iIdx, oIdx)
utt.verify_grad(metaop, [b_val, h_val, W_val], mode=self.mode)
utt.verify_grad(op, [b_val, h_val, W_val], mode=self.mode)
eps = 3e-4
eps = 3e-3
utt.verify_grad(metaop, [b_val, h_val, W_val], mode=self.mode, eps=eps)
utt.verify_grad(op, [b_val, h_val, W_val], mode=self.mode, eps=eps)
def test_sparseblockgemv_grad_1(self):
"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论