提交 3e586467 authored 作者: Nicolas Ballas's avatar Nicolas Ballas

update test_abstract_conv

上级 db1dfdea
from __future__ import absolute_import, print_function, division
import unittest
import numpy
......@@ -342,18 +341,14 @@ class TestAbstractConvNoOptim(BaseTestConv2d):
cls.filters_shapes = [(5, 1, 2, 2)]
cls.subsamples = [(1, 1), (2, 2)]
cls.filters_dilations = [(1, 1), (1, 2), (2, 1)]
cls.border_modes = ["valid", "full"]
cls.border_modes = ["valid", "half", "full"]
cls.filter_flip = [True]
cls.provide_shape = [False]
def tcase(self, i, f, s, b, flip, provide_shape, fd=(1, 1)):
o = self.get_output_shape(i, f, s, b, fd)
if (not theano.config.blas.ldflags or
not theano.config.cxx or
theano.config.mode == "FAST_COMPILE"):
raise SkipTest("Need blas to test conv2d")
mode = theano.Mode(optimizer=None)
self.run_fwd(inputs_shape=i, filters_shape=f, subsample=s,
verify_grad=True, provide_shape=provide_shape,
border_mode=b, filter_flip=flip,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论