提交 6ccc37b9 authored 作者: Frederic's avatar Frederic

fix test crash du to op name change

上级 2e4eeefc
......@@ -397,7 +397,9 @@ def build_conv_nnet2_classif(use_gpu, isize, ksize, n_batch,
# Check that GpuConv is used
topo = train.maker.fgraph.toposort()
conv_ops = (tcn.blas.GpuConv,
tcn.dnn.GpuDnnConvBase,
tcn.dnn.GpuDnnConv,
tcn.dnn.GpuDnnConvGradI,
tcn.dnn.GpuDnnConvGradW,
tcn.blas.BaseGpuCorrMM)
assert len([n for n in topo if isinstance(n.op, conv_ops)]) > 0
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论