From 0d45568bd169af2ae0ad1a22462d5720e7ab5f81 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Tue, 18 Feb 2025 14:17:52 +0100 Subject: [PATCH] Uniform validation, Add cosine_similarity validation to face swapper --- embedding_converter/src/training.py | 3 ++- face_swapper/src/training.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index b77e3a5..ad69e97 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -106,7 +106,8 @@ def create_trainer() -> Trainer: save_top_k = 3, save_last = True ) - ] + ], + val_check_interval = 10 ) diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index 89fa662..1ca38ba 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -118,7 +118,7 @@ def create_trainer() -> Trainer: save_last = True ) ], - val_check_interval = 1000, + val_check_interval = 1000 )