From 35b779b1edf409ca875b46a9fb03acc9efca19e3 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Thu, 27 Feb 2025 19:24:12 +0530 Subject: [PATCH] change pretrained models mode to eval --- face_swapper/src/training.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index 2d38f53..e8dd59f 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -30,9 +30,9 @@ class FaceSwapperTrainer(lightning.LightningModule): landmarker_path = CONFIG.get('training.model', 'landmarker_path') motion_extractor_path = CONFIG.get('training.model', 'motion_extractor_path') - self.embedder = torch.jit.load(embedder_path, map_location = 'cpu') # type:ignore[no-untyped-call] - self.landmarker = torch.jit.load(landmarker_path, map_location = 'cpu') # type:ignore[no-untyped-call] - self.motion_extractor = torch.jit.load(motion_extractor_path, map_location = 'cpu') # type:ignore[no-untyped-call] + self.embedder = torch.jit.load(embedder_path, map_location = 'cpu').eval() # type:ignore[no-untyped-call] + self.landmarker = torch.jit.load(landmarker_path, map_location = 'cpu').eval() # type:ignore[no-untyped-call] + self.motion_extractor = torch.jit.load(motion_extractor_path, map_location = 'cpu').eval() # type:ignore[no-untyped-call] self.generator = Generator() self.discriminator = Discriminator()