提交 8f75cba4 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

Allow profile argument of scan to be a bool

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