increase random tensor size (#243)

Co-authored-by: Ainur Karimov <[email protected]>
InnopolisUni · Dec 12, 2024 · 7c0d925 · 7c0d925
1 parent f814256
commit 7c0d925
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/losses/providers/torch/test_segmentation.py b/tests/unit/losses/providers/torch/test_segmentation.py
@@ -66,7 +66,7 @@ def test_iou_batch():
 
 def test__threshold():
 
-    x = torch.rand(2, 3)
+    x = torch.rand(10, 10)
     out = _threshold(x, threshold=0.5)
     assert torch.max(out) == 1
     assert torch.min(out) == 0