提交 285316c1 authored 作者: Frederic Bastien's avatar Frederic Bastien

make test faster.

上级 46db4371
...@@ -88,10 +88,10 @@ class TestConv2D(unittest.TestCase): ...@@ -88,10 +88,10 @@ class TestConv2D(unittest.TestCase):
Tests that basic convolutions work for odd and even dimensions of image and filter Tests that basic convolutions work for odd and even dimensions of image and filter
shapes, as well as rectangular images and filters. shapes, as well as rectangular images and filters.
""" """
self.validate((3,2,8,8), (4,2,5,5), 'valid') self.validate((3,2,8,8), (4,2,5,5), 'valid', verify_grad=False)
self.validate((3,2,7,5), (5,2,2,3), 'valid') self.validate((3,2,7,5), (5,2,2,3), 'valid')
self.validate((3,2,7,5), (5,2,3,2), 'valid') self.validate((3,2,7,5), (5,2,3,2), 'valid', verify_grad=False)
self.validate((3,2,8,8), (4,2,5,5), 'full') self.validate((3,2,8,8), (4,2,5,5), 'full', verify_grad=False)
self.validate((3,2,7,5), (5,2,2,3), 'full') self.validate((3,2,7,5), (5,2,2,3), 'full')
# test filter same size as input # test filter same size as input
...@@ -105,7 +105,7 @@ class TestConv2D(unittest.TestCase): ...@@ -105,7 +105,7 @@ class TestConv2D(unittest.TestCase):
""" """
self.validate((3,2,7,5), (5,2,2,3), 'valid', unroll_patch=False) self.validate((3,2,7,5), (5,2,2,3), 'valid', unroll_patch=False)
self.validate((3,2,7,5), (5,2,2,3), 'full', unroll_patch=False) self.validate((3,2,7,5), (5,2,2,3), 'full', unroll_patch=False)
self.validate((3,2,3,3), (4,2,3,3), 'valid', unroll_patch=False) self.validate((3,2,3,3), (4,2,3,3), 'valid', unroll_patch=False, verify_grad=False)
def test_unroll_special(self): def test_unroll_special(self):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论