提交 b41c0a03 authored 作者: Harm de Vries's avatar Harm de Vries

remove prints

上级 bbd9ff3f
...@@ -1459,7 +1459,6 @@ if (pool%(name)s != NULL) { cudnnDestroyPoolingDescriptor(pool%(name)s); } ...@@ -1459,7 +1459,6 @@ if (pool%(name)s != NULL) { cudnnDestroyPoolingDescriptor(pool%(name)s); }
raise NotImplementedError("Unsupported pooling model.") raise NotImplementedError("Unsupported pooling model.")
return """ return """
fprintf(stderr, "test_forward\\n");
cudnnStatus_t err; cudnnStatus_t err;
int %(out)s_dims[5]; int %(out)s_dims[5];
...@@ -1594,6 +1593,9 @@ class GpuDnnPoolGrad(DnnBase): ...@@ -1594,6 +1593,9 @@ class GpuDnnPoolGrad(DnnBase):
assert (inp_grad.ndim in [4, 5]) assert (inp_grad.ndim in [4, 5])
out = as_cuda_ndarray_variable(out) out = as_cuda_ndarray_variable(out)
assert(out.ndim in [4, 5]) assert(out.ndim in [4, 5])
assert (inp_grad.ndim == inp.dim)
assert (inp.dim == out.dim)
ws = tensor.as_tensor_variable(ws) ws = tensor.as_tensor_variable(ws)
stride = tensor.as_tensor_variable(stride) stride = tensor.as_tensor_variable(stride)
...@@ -1679,10 +1681,9 @@ if (pool%(name)s != NULL) { cudnnDestroyPoolingDescriptor(pool%(name)s); } ...@@ -1679,10 +1681,9 @@ if (pool%(name)s != NULL) { cudnnDestroyPoolingDescriptor(pool%(name)s); }
raise Exception("cudnn v1 do not support average_exc_pad") raise Exception("cudnn v1 do not support average_exc_pad")
else: else:
raise NotImplementedError("Unsupported pooling model.") raise NotImplementedError("Unsupported pooling model.")
print mode_flag
return """ return """
cudnnStatus_t err%(name)s; cudnnStatus_t err%(name)s;
//raise(SIGINT);
if (!CudaNdarray_is_c_contiguous(%(input)s)) { if (!CudaNdarray_is_c_contiguous(%(input)s)) {
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
......
...@@ -348,7 +348,7 @@ def test_pooling(): ...@@ -348,7 +348,7 @@ def test_pooling():
assert any([isinstance(node.op, AveragePoolGrad) assert any([isinstance(node.op, AveragePoolGrad)
for node in fc.maker.fgraph.toposort()]) for node in fc.maker.fgraph.toposort()])
c_out = fc(data) c_out = fc(data)
assert numpy.allclose(c_out, g_out) assert utt.assert_allclose(c_out, g_out)
def test_pooling3d(): def test_pooling3d():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论