qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-03-01 21:02:00 +01:00
Code Issues Projects Releases Wiki Activity

training.utils.preprocess_imgs: fix file shuffling in 27f43c17

Browse source
This commit is contained in:
Robert Sachunsky 2026-02-24 20:37:44 +01:00
parent 658dade0d4
commit 20a3672be3
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/eynollah/training/utils.py
View file

@ -779,9 +779,10 @@ def preprocess_imgs(config,
# override keys from call # override keys from call
config.update(kwargs) config.update(kwargs)
seed = random.random() seed = random.getstate()
random.shuffle(imgs_list, random=lambda: seed) random.shuffle(imgs_list)
random.shuffle(labs_list, random=lambda: seed) random.setstate(seed)
random.shuffle(labs_list)
# labs_list not used because stem matching more robust # labs_list not used because stem matching more robust
indexer = 0 indexer = 0