From e6ea4543608d01d3216100212e0d64b302a42e5b Mon Sep 17 00:00:00 2001
From: harisreedhar <h4harisreedhar.s.s@gmail.com>
Date: Mon, 3 Mar 2025 15:12:35 +0530
Subject: [PATCH] supress warning

---
 face_swapper/src/training.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py
index 590606c..637f094 100644
--- a/face_swapper/src/training.py
+++ b/face_swapper/src/training.py
@@ -1,5 +1,6 @@
 import configparser
 import os
+import warnings
 from typing import Tuple, cast
 
 import lightning
@@ -19,6 +20,8 @@ from .models.generator import Generator
 from .models.loss import AdversarialLoss, AttributeLoss, DiscriminatorLoss, GazeLoss, IdentityLoss, PoseLoss, ReconstructionLoss
 from .types import Batch, Embedding, OptimizerConfig, WarpTemplate
 
+warnings.filterwarnings("ignore", category=UserWarning, module="torch")
+
 CONFIG = configparser.ConfigParser()
 CONFIG.read('config.ini')