提交 8e46172a authored 作者: Frederic Bastien's avatar Frederic Bastien

Only reload the files and do not redo all the init.

上级 5ba45c50
......@@ -374,7 +374,7 @@ class GpuDnnConv(DnnBase, COp):
self.inplace = False
# Work around to reload old pickle.
# We need to find the new file name and reload c code.
self.__init__(**self._props_dict())
self.load_c_code(["dnn_base.c", "dnn_conv_base.c", "dnn_fwd.c"])
def get_op_params(self):
if self.inplace:
......@@ -639,6 +639,7 @@ class GpuDnnConvGradW(DnnBase, COp):
self.algo = config.dnn.conv.algo_bwd_filter
if not hasattr(self, 'inplace'):
self.inplace = False
self.load_c_code(["dnn_base.c", "dnn_conv_base.c", "dnn_gw.c"])
def grad(self, inp, grads):
img, top, output, desc, alpha, beta = inp
......@@ -866,7 +867,7 @@ class GpuDnnConvGradI(DnnBase, COp):
self.algo = config.dnn.conv.algo_bwd_data
if not hasattr(self, 'inplace'):
self.inplace = False
self.__init__(**self._props_dict())
self.load_c_code(["dnn_base.c", "dnn_conv_base.c", "dnn_gi.c"])
def grad(self, inp, grads):
kerns, top, output, desc, alpha, beta = inp
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论