未验证 提交 3bc034aa 编写于 作者: K Kaipeng Deng 提交者: GitHub

refine create_parameter to global_var to avoid saving to checkpoints (#1274)

上级 a30abac2
...@@ -229,11 +229,7 @@ class IouLoss(object): ...@@ -229,11 +229,7 @@ class IouLoss(object):
return x1, y1, x2, y2 return x1, y1, x2, y2
def _create_tensor_from_numpy(self, numpy_array): def _create_tensor_from_numpy(self, numpy_array):
paddle_array = fluid.layers.create_parameter( paddle_array = fluid.layers.create_global_var(
attr=ParamAttr(), shape=numpy_array.shape, value=0., dtype=numpy_array.dtype)
shape=numpy_array.shape, fluid.layers.assign(numpy_array, paddle_array)
dtype=numpy_array.dtype,
default_initializer=NumpyArrayInitializer(numpy_array))
paddle_array.stop_gradient = True
paddle_array.persistable = False
return paddle_array return paddle_array
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册