提交 f4fb1e57 authored 作者: ChienliMa's avatar ChienliMa

Fit Profile_maker.create() with param storage_map

上级 c5949e3e
......@@ -43,10 +43,9 @@ AddConfigVar('ProfileMode.profile_memory',
class Profile_Maker(FunctionMaker):
# storage_map does not work in Profile_Maker
# I just add this argument to fit the interface. -- ChienliMa 03.June.2015
def create(self, input_storage=None, trustme=False, storage_map=None):
ret = super(Profile_Maker, self).create(input_storage, trustme)
ret = super(Profile_Maker, self).create(input_storage, storage_map,
trustme)
if (hasattr(theano, 'sandbox') and
hasattr(theano.sandbox, 'cuda') and
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论