提交 c964257d authored 作者: Razvan Pascanu's avatar Razvan Pascanu

accept ints as well since many people write profile=1

上级 56366926
......@@ -355,7 +355,7 @@ class Scan(PureOp):
self.outputs[:slices] ]
wrapped_outputs += self.outputs[slices:]
profile = None
if (theano.config.profile or (isinstance(self.profile, (str,bool))
if (theano.config.profile or (isinstance(self.profile, (str,bool,int))
and self.profile)):
if isinstance(self.profile, str):
profile = ScanProfileStats(name = self.profile)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论