Skip to content

Commit 6044340

Browse files
author
Anna Grebneva
authored
Fixed setting of default_layout for model with non-standard channel number (#3151)
1 parent 6e6fca6 commit 6044340

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/launcher/openvino_launcher.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def try_to_set_default_layout(self):
152152
if channel_dim in [3, -1]:
153153
self.default_layout = 'NHWC'
154154
return
155-
if shape[-1] in [1, 3, 4]:
155+
if shape[-1] in [1, 2, 3, 4, 6, 9]:
156156
self.default_layout = 'NHWC'
157157
return
158158
self.default_layout = 'NCHW'

0 commit comments

Comments
 (0)