Modify unit tests

openvinotoolkit · Jun 26, 2023 · 3c33ab0 · 3c33ab0
1 parent 36ccfb1
commit 3c33ab0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/unit/algorithms/detection/adapters/mmdet/models/heads/test_custom_dino_head.py b/tests/unit/algorithms/detection/adapters/mmdet/models/heads/test_custom_dino_head.py
@@ -171,10 +171,10 @@ def test_forward_train(self):
     @e2e_pytest_unit
     def test_simple_test_bboxes(self):
         feats = [
-            torch.randn([2, 256, 100, 134]),
-            torch.randn([2, 256, 50, 67]),
-            torch.randn([2, 256, 25, 34]),
-            torch.randn([2, 256, 13, 17]),
+            torch.zeros([2, 256, 100, 134]),
+            torch.zeros([2, 256, 50, 67]),
+            torch.zeros([2, 256, 25, 34]),
+            torch.zeros([2, 256, 13, 17]),
         ]
         img_metas = [
             {