提交 26638e9c 编写于 作者: C chengduoZH

fix batch_norm parameter share

上级 99e3d1ec
......@@ -2421,6 +2421,11 @@ class BatchNormLayer(LayerBase):
psize = self.calc_parameter_size(image_conf)
dims = [1, psize]
self.inputs[1].parameter_name = self.inputs[0].parameter_name.split('.')[0] + '.' + \
self.inputs[1].parameter_name.split('.')[1]
self.inputs[2].parameter_name = self.inputs[0].parameter_name.split('.')[0] + '.' + \
self.inputs[2].parameter_name.split('.')[1]
self.create_input_parameter(0, psize)
self.create_input_parameter(1, psize, dims)
self.create_input_parameter(2, psize, dims)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册