提交 a5fe8e38 authored 作者: --global's avatar --global

In __setstate__ use flag value as default value

上级 349378f1
......@@ -457,7 +457,7 @@ class GpuDnnConv(DnnBase, COp):
if hasattr(self, 'workmem'):
self.algo = self.workmem
else:
self.algo = 'none'
self.algo = config.dnn.conv.algo_fwd
if not hasattr(self, 'inplace'):
self.inplace = False
......@@ -724,7 +724,7 @@ class GpuDnnConvGradW(DnnBase, COp):
if hasattr(self, 'workmem'):
self.algo = self.workmem
else:
self.algo = 'none'
self.algo = config.dnn.conv.algo_bwd
if not hasattr(self, 'inplace'):
self.inplace = False
......@@ -911,7 +911,7 @@ class GpuDnnConvGradI(DnnBase, COp):
if hasattr(self, 'workmem'):
self.algo = self.workmem
else:
self.algo = 'none'
self.algo = config.dnn.conv.algo_bwd
if not hasattr(self, 'inplace'):
self.inplace = False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论