From 257e5e56a4d0f45a58fd24e6b5ab984257494fc9 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 24 Feb 2025 09:39:54 +0100 Subject: [PATCH] Rename split ratio --- embedding_converter/src/training.py | 4 ++-- face_swapper/src/training.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index 93e3ac2..14358fa 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -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 diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index f3b132a..e71d909 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -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