未验证 提交 84cc1197 编写于 作者: S SunAhong1993 提交者: GitHub

Merge pull request #384 from SunAhong1993/syf_caffe

0.8.4
__version__ = "0.8.1" __version__ = "0.8.4"
from .core.program import PaddleProgram from .core.program import PaddleProgram
......
...@@ -226,7 +226,7 @@ class CaffeOpMapper(OpMapper): ...@@ -226,7 +226,7 @@ class CaffeOpMapper(OpMapper):
data.append( data.append(
np.zeros([output_c, input_c, kernel[0], kernel[1]]).astype( np.zeros([output_c, input_c, kernel[0], kernel[1]]).astype(
'float32')) 'float32'))
data.append(np.zeros([output_c, ])).astype('float32') data.append(np.zeros([output_c, ]).astype('float32'))
else: else:
data = self.adjust_parameters(node) data = self.adjust_parameters(node)
self.weights[node.layer_name + '_weights'] = data[0] self.weights[node.layer_name + '_weights'] = data[0]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册