提交 26454be6 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix a test

上级 3ab42636
...@@ -1551,7 +1551,7 @@ def test_without_dnn_batchnorm_train_without_running_averages(): ...@@ -1551,7 +1551,7 @@ def test_without_dnn_batchnorm_train_without_running_averages():
f_abstract = theano.function([x, scale, bias, dy], f_abstract = theano.function([x, scale, bias, dy],
[out_abstract, x_mean_abstract, x_invstd_abstract] + [out_abstract, x_mean_abstract, x_invstd_abstract] +
grads_abstract, grads_abstract,
mode=mode_with_gpu) mode=mode_with_gpu.excluding('cudnn'))
# check if the abstract Ops have been replaced # check if the abstract Ops have been replaced
assert not any([isinstance(n.op, dnn.GpuDnnBatchNorm) assert not any([isinstance(n.op, dnn.GpuDnnBatchNorm)
for n in f_abstract.maker.fgraph.toposort()]) for n in f_abstract.maker.fgraph.toposort()])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论