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

Minor change

Browse Source
This commit is contained in:
henryruhs
2025-03-12 20:18:33 +01:00
parent bf7bbc2550
commit 0732924f1e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/training.py
+2 -2
View File
@@ -124,8 +124,8 @@ class FaceSwapperTrainer(LightningModule):
self.untoggle_optimizer(generator_optimizer)
self.toggle_optimizer(masker_optimizer)
target_attribute = generator_output_attributes[-1].detach()
mask_tensor = self.masker(target_tensor, target_attribute)
target_attribute = generator_output_attributes[-1]
mask_tensor = self.masker(target_tensor, target_attribute.detach())
mask_loss = self.mask_loss(target_tensor, mask_tensor)
self.manual_backward(mask_loss)