提交 ecea6d5b 编写于 作者: 李寅

Merge branch 'feature_wuch' into 'master'

fix tf converter bug

See merge request !445
......@@ -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.
先完成此消息的编辑!
想要评论请 注册