From 9df29f8a229b48fe23688df629984e3f755e7934 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Wed, 26 Mar 2025 16:54:05 +0530 Subject: [PATCH] changes --- face_swapper/src/training.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index 7c72136..7507f91 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -192,8 +192,8 @@ def combine_datasets() -> Dataset[Tensor]: dataset_config.set('training.dataset', key, value) datasets.append(DynamicDataset(dataset_config)) - combined_dataset = ConcatDataset(datasets) - return combined_dataset + combine_dataset = ConcatDataset(datasets) + return combine_dataset def create_trainer() -> Trainer: