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

Rename variable of EmbeddingConverterTrainer

Browse Source
This commit is contained in:
henryruhs
2025-02-17 23:25:19 +01:00
parent 254f3efe68
commit 999f2c9cbe
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/training.py
+3 -3
View File
@@ -113,7 +113,7 @@ def create_trainer() -> Trainer:
def train() -> None:
trainer = create_trainer()
training_loader, validation_loader = create_loaders()
embedding_converter = EmbeddingConverterTrainer()
embedding_converter_trainer = EmbeddingConverterTrainer()
tuner = Tuner(trainer)
tuner.lr_find(embedding_converter, training_loader, validation_loader)
trainer.fit(embedding_converter, training_loader, validation_loader)
tuner.lr_find(embedding_converter_trainer, training_loader, validation_loader)
trainer.fit(embedding_converter_trainer, training_loader, validation_loader)