提交 85dbe94c 编写于 作者: S SunAhong1993

fix the prelu

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