提交 5fe08d1f 编写于 作者: W wuchenghui

fix tf converter bug

上级 09c17afa
......@@ -406,7 +406,7 @@ class TFConverter(object):
input_channels = weight_tensor_value.shape[2]
# HWIO -> OIHW
weight_tensor_value = weight_tensor_value.transpose(3, 2, 0, 1)
if input_shape[2] > 16 and input_shape[3] > 16:
if input_shape[1] > 16 and input_shape[2] > 16:
G = np.array([
[1.0, 0.0, 0.0],
[-2.0 / 9, -2.0 / 9, -2.0 / 9],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册