CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-06-25 07:59:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Uniform validation, Add cosine_similarity validation to face swapper

Browse Source
This commit is contained in:
henryruhs
2025-02-18 14:14:23 +01:00
parent c161da2f25
commit 28977d37d6
2 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/training.py
+5 -5
View File
@@ -38,9 +38,9 @@ class EmbeddingConverterTrainer(lightning.LightningModule):
def validation_step(self, batch : Batch, batch_index : int) -> Tensor:
source_tensor, target_tensor = batch
output_tensor = self(source_tensor)
loss_validation = self.mse_loss(output_tensor, target_tensor)
self.log('loss_validation', loss_validation, prog_bar = True)
return loss_validation
validation = self.mse_loss(output_tensor, target_tensor)
self.log('validation', validation, prog_bar = True)
return validation
def configure_optimizers(self) -> Any:
learning_rate = CONFIG.getfloat('training.trainer', 'learning_rate')
@@ -94,6 +94,7 @@ def create_trainer() -> Trainer:
return Trainer(
logger = logger,
log_every_n_steps = 10,
max_epochs = trainer_max_epochs,
callbacks =
[
@@ -106,8 +107,7 @@ def create_trainer() -> Trainer:
save_last = True
)
],
enable_progress_bar = True,
log_every_n_steps = 2
val_check_interval = 1000
)