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