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

lr only in optimizer

Browse Source
This commit is contained in:
henryruhs
2025-02-21 18:40:11 +01:00
parent d153c68813
commit 575f215408
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/training.py
-2
View File
@@ -24,7 +24,6 @@ class EmbeddingConverterTrainer(lightning.LightningModule):
super(EmbeddingConverterTrainer, self).__init__()
source_path = CONFIG.get('training.model', 'source_path')
target_path = CONFIG.get('training.model', 'target_path')
learning_rate = CONFIG.getfloat('training.trainer', 'learning_rate')
self.embedding_converter = EmbeddingConverter()
self.source_embedder = torch.jit.load(source_path) # type:ignore[no-untyped-call]
@@ -32,7 +31,6 @@ class EmbeddingConverterTrainer(lightning.LightningModule):
self.source_embedder.eval()
self.target_embedder.eval()
self.mse_loss = nn.MSELoss()
self.lr = learning_rate
def forward(self, source_embedding : Embedding) -> Embedding:
return self.embedding_converter(source_embedding)