Skip to content

Commit 6dea63c

Browse files
authored
AC: fix data providing if launcher with dyn batch (#3122)
1 parent d15c492 commit 6dea63c

File tree

1 file changed

+1
-1
lines changed
  • tools/accuracy_checker/openvino/tools/accuracy_checker

1 file changed

+1
-1
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -666,7 +666,7 @@ def sava_subset(self):
666666
yaml.safe_dump(identifiers, sf)
667667

668668
def __getitem__(self, item):
669-
if self.batch is None:
669+
if self.batch is None or self._batch <= 0:
670670
self.batch = 1
671671
if self.size <= item * self.batch:
672672
raise IndexError

0 commit comments

Comments
 (0)