supress warning

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')