From f19908ccd66d2cfe1a56abc54e1dcdf0a1f9fd11 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 24 Feb 2025 09:26:46 +0100 Subject: [PATCH] Fix spacing for lambda --- embedding_converter/src/dataset.py | 2 +- face_swapper/src/dataset.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/embedding_converter/src/dataset.py b/embedding_converter/src/dataset.py index f9e074c..62a8fae 100644 --- a/embedding_converter/src/dataset.py +++ b/embedding_converter/src/dataset.py @@ -30,6 +30,6 @@ class DynamicDataset(Dataset[Tensor]): transforms.Resize((112, 112), interpolation = transforms.InterpolationMode.BICUBIC), transforms.ColorJitter(brightness = 0.2, contrast = 0.2, saturation = 0.2, hue = 0.1), transforms.ToTensor(), - transforms.Lambda(lambda temp_tensor : temp_tensor[[2, 1, 0], :, :]), + transforms.Lambda(lambda temp_tensor: temp_tensor[[2, 1, 0], :, :]), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ]) diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py index 61bc4a3..017827f 100644 --- a/face_swapper/src/dataset.py +++ b/face_swapper/src/dataset.py @@ -35,7 +35,7 @@ class DynamicDataset(Dataset[Tensor]): transforms.ColorJitter(brightness = 0.2, contrast = 0.2, saturation = 0.2, hue = 0.1), transforms.RandomAffine(4, translate = (0.01, 0.01), scale = (0.98, 1.02), shear = (1, 1)), transforms.ToTensor(), - transforms.Lambda(lambda temp_tensor : temp_tensor[[2, 1, 0], :, :]), + transforms.Lambda(lambda temp_tensor: temp_tensor[[2, 1, 0], :, :]), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ])