CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Simplify sizes

Browse Source
This commit is contained in:
henryruhs
2025-02-21 09:11:43 +01:00
parent 251e610f0e
commit 15ee6fa763
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/training.py
+4 -3
View File
@@ -83,9 +83,10 @@ def create_loaders(dataset : Dataset[Tensor]) -> Tuple[DataLoader[Tensor], DataL
def split_dataset(dataset : Dataset[Tensor]) -> Tuple[Dataset[Tensor], Dataset[Tensor]]:
loader_split_ratio = CONFIG.getfloat('training.loader', 'split_ratio')
training_size = int(loader_split_ratio * len(dataset)) # type:ignore[operator, arg-type]
validation_size = len(dataset) - training_size # type:ignore[arg-type]
training_dataset, validate_dataset = random_split(dataset, [training_size, validation_size])
dataset_size = len(dataset)
training_size = dataset_size * loader_split_ratio
validation_size = dataset_size - training_size
training_dataset, validate_dataset = random_split(dataset, [ training_size, validation_size ])
return training_dataset, validate_dataset