diff --git a/face_swapper/README.md b/face_swapper/README.md
index e80e484..7a74819 100644
--- a/face_swapper/README.md
+++ b/face_swapper/README.md
@@ -94,7 +94,7 @@ strategy = auto
 precision = 16-mixed
 logger_path = .logs
 logger_name = face_swapper
-preview_frequency = 250
+preview_frequency = 100
 ```
 
 ```
diff --git a/face_swapper/src/helper.py b/face_swapper/src/helper.py
index 695d541..b1377d5 100644
--- a/face_swapper/src/helper.py
+++ b/face_swapper/src/helper.py
@@ -39,7 +39,6 @@ def calc_embedding(embedder : EmbedderModule, input_tensor : Tensor, padding : P
 
 
 def overlay_mask(input_tensor : Tensor, input_mask : Mask) -> Tensor:
-	input_mask = input_mask.mean(dim = 1, keepdim = True)
 	overlay_tensor = torch.zeros(*input_tensor.shape, dtype = input_tensor.dtype, device = input_tensor.device)
 	overlay_tensor[:, 2, :, :] = 1
 	input_mask = input_mask.repeat(1, 3, 1, 1).clamp(0, 0.8)