提交 c237dc5c authored 作者: James Bergstra's avatar James Bergstra

tests/test_conv_cuda_ndarray - 80char formatting

上级 36617736
...@@ -177,8 +177,6 @@ def _params_allgood(ishape, kshape, mode, subsample=(1,1), img_stride=(1,1), ...@@ -177,8 +177,6 @@ def _params_allgood(ishape, kshape, mode, subsample=(1,1), img_stride=(1,1),
print "max absolute diff:",diffabs.max(),"avg abs diff:",numpy.average(diffabs) print "max absolute diff:",diffabs.max(),"avg abs diff:",numpy.average(diffabs)
print "median abs diff:", numpy.median(diffabs), "nb close:",nb_close, "/", diff.size print "median abs diff:", numpy.median(diffabs), "nb close:",nb_close, "/", diff.size
print "max relatif diff:",pr_diff.max(), "avg rel diff:", numpy.average(pr_diff) print "max relatif diff:",pr_diff.max(), "avg rel diff:", numpy.average(pr_diff)
print rval
if not rval and print_!=False: if not rval and print_!=False:
if npy_img.shape[0]>5: if npy_img.shape[0]>5:
print "img",npy_img[0] print "img",npy_img[0]
...@@ -207,9 +205,19 @@ def exec_conv(version, shapes, verbose, random, mode, print_=None, rtol=1e-5, on ...@@ -207,9 +205,19 @@ def exec_conv(version, shapes, verbose, random, mode, print_=None, rtol=1e-5, on
for id,(ishape, kshape, subshape, istride, kstride) in enumerate(shapes): for id,(ishape, kshape, subshape, istride, kstride) in enumerate(shapes):
ret=False ret=False
try: try:
ret = _params_allgood(ishape, kshape, mode, ret = _params_allgood(ishape,
subsample=subshape, img_stride=istride, kern_stride=kstride, kshape,
version=ver, verbose=verbose, random=random, id=id,print_=print_,rtol=rtol,ones=ones) mode,
subsample=subshape,
img_stride=istride,
kern_stride=kstride,
version=ver,
verbose=verbose,
random=random,
id=id,
print_=print_,
rtol=rtol,
ones=ones)
except Exception, e: except Exception, e:
print ver, id,(ishape, kshape, subshape, istride, kstride) print ver, id,(ishape, kshape, subshape, istride, kstride)
print e print e
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论