提交 b7d294bf 编写于 作者: S SunAhong1993

fix the prelu

上级 85dbe94c
...@@ -244,9 +244,11 @@ class CaffeDecoder(object): ...@@ -244,9 +244,11 @@ class CaffeDecoder(object):
if len(blob.shape.dim): if len(blob.shape.dim):
dims = blob.shape.dim dims = blob.shape.dim
if layer.type == 'PReLU': if layer.type == 'PReLU':
c_o, c_i, h, w = map(int, [1] + list(dims) + [1]* (3 - len(dims))) c_o, c_i, h, w = map(int, [1] + \
list(dims) + [1]* (3 - len(dims)))
else: else:
c_o, c_i, h, w = map(int, [1] * (4 - len(dims)) + list(dims)) c_o, c_i, h, w = map(int, [1] * (4 - len(dims)) \
+ list(dims))
else: else:
c_o = blob.num c_o = blob.num
c_i = blob.channels c_i = blob.channels
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册