提交 23a06193 authored 作者: Alexander Matyasko's avatar Alexander Matyasko

Support more debuggers for breakpoint

上级 2c6adccc
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import numpy as np import numpy as np
import pdb import imp
import theano import theano
from theano.gof import Op, Apply from theano.gof import Op, Apply
...@@ -116,7 +116,16 @@ class PdbBreakpoint(Op): ...@@ -116,7 +116,16 @@ class PdbBreakpoint(Op):
print("Their contents can be altered and, when execution") print("Their contents can be altered and, when execution")
print("resumes, the updated values will be used.") print("resumes, the updated values will be used.")
print("-------------------------------------------------") print("-------------------------------------------------")
pdb.set_trace()
if imp.find_module('pudb'):
import pudb
pudb.set_trace()
elif imp.find_module('ipdb'):
import ipdb
ipdb.set_trace()
else:
import pdb
pdb.set_trace()
# Take the new values in monitored, cast them back to their # Take the new values in monitored, cast them back to their
# original type and store them in the output_storage # original type and store them in the output_storage
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论