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

typo

This commit is contained in:
Damian Stewart 2023-06-14 09:28:40 +02:00
parent e4872fdc0c
commit bd89ad96d2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data/data_loader.py
View File

@ -124,7 +124,7 @@ class DataLoaderMultiAspect():
continue
runts = bucket[-runt_count:]
del bucket[-runt_count:]
matching_default_bucket_key = [DEFAULT_BATCH_ID, key[1], key[2]]
matching_default_bucket_key = (DEFAULT_BATCH_ID, key[1], key[2])
buckets[matching_default_bucket_key].extend(runts)
# handle remaining runts by randomly duplicating items