CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

add expression loss to log

Browse Source
This commit is contained in:
harisreedhar
2025-03-20 11:01:54 +05:30
parent b6a2734622
commit 35afb426b7
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/training.py
+1
View File
@@ -130,6 +130,7 @@ class FaceSwapperTrainer(LightningModule):
self.log('reconstruction_loss', reconstruction_loss)
self.log('identity_loss', identity_loss)
self.log('pose_loss', pose_loss)
self.log('expression_loss', expression_loss)
self.log('gaze_loss', gaze_loss)
self.log('mask_loss', mask_loss)
return generator_loss