Mirrors
/
EveryDream-trainer
mirror of https://github.com/victorchall/EveryDream-trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixed shuffle

This commit is contained in:
Victor Hall 2022-10-26 02:13:59 -04:00
parent 0ed7b0b007
commit cbca4d13ce
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ldm/data/every_dream.py
View File

@ -31,13 +31,13 @@ class EveryDreamBatch(Dataset):
image_path = os.path.join(class_path, file_path)
self.image_paths.append(image_path)
import random
# improve multi-class training by mixing order of training set, avoid training on one class N times in a row
# if trainer crashes between epochs and you resume at least it isn't heavily biasing early folders in training set
#self.image_paths = random.Random(555).shuffle(self.image_paths)
print(self.image_paths)
import random
random.Random(555).shuffle(self.image_paths)
print(f"**** Loaded {len(self.image_paths)} images fromt {self.data_root}")
# self._length = len(self.image_paths)
self.num_images = len(self.image_paths)
self._length = self.num_images