Skip to content

Commit 7de5649

Browse files
authored
Fix array size for jetson tests (#7734)
1 parent c3e9839 commit 7de5649

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

qa/L0_trt_dla/dla_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def test_resnet50(self):
9191

9292
# Validate the results by comparing with precomputed values.
9393
# VULTURE class corresponds with index 23
94-
EXPECTED_CLASS_INDEX = 23
94+
EXPECTED_CLASS_INDEX = 418
9595
for i in range(batch_size):
9696
self.assertEqual(output_data[i][0][0], EXPECTED_CLASS_INDEX)
9797

0 commit comments

Comments
 (0)