提交 7e22b0d2 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix indent.

上级 71027a03
...@@ -80,14 +80,12 @@ def _params_allgood(ishape, kshape, mode, subsample=(1,1), img_stride=(1,1), ker ...@@ -80,14 +80,12 @@ def _params_allgood(ishape, kshape, mode, subsample=(1,1), img_stride=(1,1), ker
t2 = None t2 = None
rval = True rval = True
try: try:
if True:
t0 = time.time() t0 = time.time()
cpuval = py_conv_scipy(npy_img, npy_kern, mode, subsample) cpuval = py_conv_scipy(npy_img, npy_kern, mode, subsample)
t1 = time.time() t1 = time.time()
i = cuda_tensor4() i = cuda_tensor4()
k = cuda_tensor4() k = cuda_tensor4()
op = theano.sandbox.cuda.blas.GpuConv(border_mode=mode,subsample=subsample, version=version, verbose=verbose)(i,k) op = theano.sandbox.cuda.blas.GpuConv(border_mode=mode,subsample=subsample, version=version, verbose=verbose)(i,k)
# imshp=ishape[1:],kshp=kshape[2:],nkern=kshape[0],bsize=ishape[0])
f=theano.function([i,k],op) f=theano.function([i,k],op)
gpuval = f(img,kern) gpuval = f(img,kern)
t2 = time.time() t2 = time.time()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论