提交 4c68405c 编写于 作者: Q qingqing01

Fix name in test_lookup_table

上级 544458e0
...@@ -4,7 +4,7 @@ from op_test_util import OpTestMeta ...@@ -4,7 +4,7 @@ from op_test_util import OpTestMeta
from gradient_checker import GradientChecker, create_op from gradient_checker import GradientChecker, create_op
class TestSigmoidOp(unittest.TestCase): class TestLookupTableOp(unittest.TestCase):
__metaclass__ = OpTestMeta __metaclass__ = OpTestMeta
def setUp(self): def setUp(self):
...@@ -15,7 +15,7 @@ class TestSigmoidOp(unittest.TestCase): ...@@ -15,7 +15,7 @@ class TestSigmoidOp(unittest.TestCase):
self.outputs = {'Out': table[ids]} self.outputs = {'Out': table[ids]}
class TestSigmoidGradOp(GradientChecker): class TestLookupTableGradOp(GradientChecker):
def test_grad(self): def test_grad(self):
op = create_op('lookup_table') op = create_op('lookup_table')
table = np.random.random((17, 31)).astype('float32') table = np.random.random((17, 31)).astype('float32')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册