提交 7848292a authored 作者: affanv14's avatar affanv14

flake8 fix

上级 687f4b20
...@@ -2223,7 +2223,7 @@ class Cudnn_grouped_conv(unittest.TestCase): ...@@ -2223,7 +2223,7 @@ class Cudnn_grouped_conv(unittest.TestCase):
desc = dnn.GpuDnnConvDesc(border_mode='valid', subsample=(1, 1), dilation=(1, 1), desc = dnn.GpuDnnConvDesc(border_mode='valid', subsample=(1, 1), dilation=(1, 1),
conv_mode='conv', precision=set_precision(theano.config.floatX))(kern.shape) conv_mode='conv', precision=set_precision(theano.config.floatX))(kern.shape)
return dnn.GpuDnnConvGradW(num_groups=groups)(img, out, kern, desc, alpha=0.75, return dnn.GpuDnnConvGradW(num_groups=groups)(img, out, kern, desc, alpha=0.75,
beta=-1.0) beta=-1.0)
utt.verify_grad(dconvw, [img, kern, top], eps=1e-3, mode=mode_with_gpu) utt.verify_grad(dconvw, [img, kern, top], eps=1e-3, mode=mode_with_gpu)
...@@ -2262,6 +2262,6 @@ class Cudnn_grouped_conv(unittest.TestCase): ...@@ -2262,6 +2262,6 @@ class Cudnn_grouped_conv(unittest.TestCase):
desc = dnn.GpuDnnConvDesc(border_mode='valid', subsample=(1, 1), dilation=(1, 1), desc = dnn.GpuDnnConvDesc(border_mode='valid', subsample=(1, 1), dilation=(1, 1),
conv_mode='conv', precision=set_precision(theano.config.floatX))(kern.shape) conv_mode='conv', precision=set_precision(theano.config.floatX))(kern.shape)
return dnn.GpuDnnConvGradI(num_groups=groups)(kern, out, img, desc, alpha=-1.0, return dnn.GpuDnnConvGradI(num_groups=groups)(kern, out, img, desc, alpha=-1.0,
beta=0.0) beta=0.0)
utt.verify_grad(dconvi, [img, kern, top], eps=1e-3, mode=mode_with_gpu) utt.verify_grad(dconvi, [img, kern, top], eps=1e-3, mode=mode_with_gpu)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论