提交 6573f824 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Change the version test to alleviate update problems.

上级 b9894b35
......@@ -473,7 +473,7 @@ class test_DnnSoftMax(test_nnet.test_SoftMax):
utt.verify_grad(softmax_op, [x_val])
# Gradient is broken for (n, c, 1, 1) in v3 rc1
if cuda.dnn.version() == (2000, 2000):
if cuda.dnn.version() != (3000, 3000):
utt.verify_grad(softmax_op, [x_val2])
def test_cudnn_softmax_grad_opt(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论