Skip to content
Snippets Groups Projects
Unverified Commit 61bc2107 authored by Yaman Umuroglu's avatar Yaman Umuroglu Committed by GitHub
Browse files

[Test] use 100 instead of 1000 labels for labelselect

1000 labels with low folding creates very large, long-running test cases
parent e718be91
No related branches found
No related tags found
No related merge requests found
......@@ -83,7 +83,7 @@ def prepare_inputs(input_tensor, idt):
@pytest.mark.parametrize("idt", [DataType.UINT8, DataType.UINT16, DataType.INT16])
# labels
@pytest.mark.parametrize("labels", [10, 1000])
@pytest.mark.parametrize("labels", [10, 100])
# folding
@pytest.mark.parametrize("fold", [-1, 2, 10])
# number of top labels to select
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment