提交 0ddda8ce authored 作者: Frederic Bastien's avatar Frederic Bastien

fix pydotprint when the mode is profilemode but the fct have not been called.

上级 b6c266e6
......@@ -375,7 +375,7 @@ def pydotprint(fct, outfile=os.path.join(config.compiledir,'theano.pydotprint.pn
blue ellipse are output of the graph
grey ellipse are var generated by the graph that are not output and are not used.
"""
if not isinstance(mode,ProfileMode):
if not isinstance(mode,ProfileMode) or not mode.fct_call.has_key(fct):
mode=None
try:
import pydot as pd
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论