CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-06-25 07:59:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Rename split ratio

Browse Source
This commit is contained in:
henryruhs
2025-02-24 09:39:54 +01:00
parent f19908ccd6
commit 257e5e56a4
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/training.py
+2 -2
View File
@@ -79,9 +79,9 @@ 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')
split_ratio = CONFIG.getfloat('training.loader', 'split_ratio')
dataset_size = len(dataset) # type:ignore[arg-type]
training_size = int(dataset_size * loader_split_ratio)
training_size = int(dataset_size * split_ratio)
validation_size = int(dataset_size - training_size)
training_dataset, validate_dataset = random_split(dataset, [ training_size, validation_size ])
return training_dataset, validate_dataset
face_swapper/src/training.py
+2 -2
View File
@@ -128,9 +128,9 @@ 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')
split_ratio = CONFIG.getfloat('training.loader', 'split_ratio')
dataset_size = len(dataset) # type:ignore[arg-type]
training_size = int(dataset_size * loader_split_ratio)
training_size = int(dataset_size * split_ratio)
validation_size = int(dataset_size - training_size)
training_dataset, validate_dataset = random_split(dataset, [ training_size, validation_size ])
return training_dataset, validate_dataset