forked from mindspore-Ecosystem/mindspore
!26334 [MD] fix occasional fail in random rampler ut
Merge pull request !26334 from liyong126/fix_random_sampler
This commit is contained in:
commit
7cdef3da6e
|
@ -48,6 +48,7 @@ def test_sequential_sampler(print_res=False):
|
||||||
|
|
||||||
|
|
||||||
def test_random_sampler(print_res=False):
|
def test_random_sampler(print_res=False):
|
||||||
|
ds.config.set_seed(1234)
|
||||||
manifest_file = "../data/dataset/testManifestData/test5trainimgs.json"
|
manifest_file = "../data/dataset/testManifestData/test5trainimgs.json"
|
||||||
map_ = {(172876, 0): 0, (54214, 0): 1, (54214, 1): 2, (173673, 0): 3, (64631, 1): 4}
|
map_ = {(172876, 0): 0, (54214, 0): 1, (54214, 1): 2, (173673, 0): 3, (64631, 1): 4}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue