提交 4ff6f281 authored 作者: Frederic's avatar Frederic

Speed up nanguardmode

上级 077accbe
......@@ -207,7 +207,7 @@ class NanGuardMode(Mode):
for j, x in enumerate(flatten(outputs)):
do_check_on(x, node, fn, False)
wrap_linker = theano.gof.WrapLinkerMany([theano.gof.OpWiseCLinker()],
[nan_check])
wrap_linker = theano.gof.WrapLinker([theano.gof.OpWiseCLinker()],
nan_check)
super(NanGuardMode, self).__init__(wrap_linker,
optimizer=theano.config.optimizer)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论