diff --git a/embedding_converter/src/dataset.py b/embedding_converter/src/dataset.py
index 05595eb..5680d59 100644
--- a/embedding_converter/src/dataset.py
+++ b/embedding_converter/src/dataset.py
@@ -28,5 +28,5 @@ class StaticDataset(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.Normalize(0.5, 0.5)
+			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 b0cd6db..87bb5f0 100644
--- a/face_swapper/src/dataset.py
+++ b/face_swapper/src/dataset.py
@@ -34,7 +34,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.Normalize(0.5, 0.5)
+			transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
 		])
 
 	def prepare_different_batch(self, source_image_path : str) -> Batch: