fix occasional failed in random sampler

2021-11-16 09:53:38 +08:00 · 2021-11-16 09:53:38 +08:00 · 9663884db3
parent 5af8572cd7
commit 9663884db3
1 changed files with 1 additions and 0 deletions
--- a/tests/ut/python/dataset/test_sampler.py
+++ b/tests/ut/python/dataset/test_sampler.py
@ -48,6 +48,7 @@ def test_sequential_sampler(print_res=False):


 def test_random_sampler(print_res=False):
+    ds.config.set_seed(1234)
    manifest_file = "../data/dataset/testManifestData/test5trainimgs.json"
    map_ = {(172876, 0): 0, (54214, 0): 1, (54214, 1): 2, (173673, 0): 3, (64631, 1): 4}