From 56be3f0b9b76048707c9ca4639464675884db58e Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sat, 1 Mar 2025 16:14:07 +0100 Subject: [PATCH] Rename to validation_score --- embedding_converter/src/training.py | 6 +++--- face_swapper/src/training.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index 0be3ce7..1fa0009 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -46,9 +46,9 @@ class EmbeddingConverterTrainer(lightning.LightningModule): with torch.no_grad(): source_embedding = self.source_embedder(batch) output_embedding = self(source_embedding) - validation = (nn.functional.cosine_similarity(source_embedding, output_embedding).mean() + 1) * 0.5 - self.log('validation', validation, prog_bar = True) - return validation + validation_score = (nn.functional.cosine_similarity(source_embedding, output_embedding).mean() + 1) * 0.5 + self.log('validation_score', validation_score, prog_bar = True) + return validation_score def configure_optimizers(self) -> OptimizerConfig: learning_rate = CONFIG.getfloat('training.trainer', 'learning_rate') diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index e8dd59f..abaf6d1 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -129,9 +129,9 @@ class FaceSwapperTrainer(lightning.LightningModule): source_embedding = calc_embedding(self.embedder, source_tensor, (0, 0, 0, 0)) output_tensor = self.generator(source_embedding, target_tensor) output_embedding = calc_embedding(self.embedder, output_tensor, (0, 0, 0, 0)) - validation = (nn.functional.cosine_similarity(source_embedding, output_embedding).mean() + 1) * 0.5 - self.log('validation', validation) - return validation + validation_score = (nn.functional.cosine_similarity(source_embedding, output_embedding).mean() + 1) * 0.5 + self.log('validation_score', validation_score, prog_bar = True) + return validation_score def generate_preview(self, source_tensor : Tensor, target_tensor : Tensor, output_tensor : Tensor) -> None: preview_limit = 8