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

Update documentation

This commit is contained in:
Joel Holdbrooks 2023-01-29 17:47:10 -08:00
parent c0ec46c030
commit 3fe335f328
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
data/data_loader.py
View File

@ -27,10 +27,11 @@ class DataLoaderMultiAspect():
""" """
Data loader for multi-aspect-ratio training and bucketing Data loader for multi-aspect-ratio training and bucketing
data_root: root folder of training data image_train_items: list of `lImageTrainItem` objects
seed: random seed
batch_size: number of images per batch batch_size: number of images per batch
""" """
def __init__(self, image_train_items, seed=555, batch_size=1): def __init__(self, image_train_items: list[ImageTrainItem], seed=555, batch_size=1):
self.seed = seed self.seed = seed
self.batch_size = batch_size self.batch_size = batch_size
# Prepare data # Prepare data