CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-07-28 16:08:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

Extend config

Browse Source
This commit is contained in:
henryruhs
2025-03-12 16:21:47 +01:00
parent 8b465fce03
commit 2af4f2c8ab
6 changed files with 24 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/training.py
+5 -3
View File
@@ -208,18 +208,20 @@ def split_dataset(dataset : Dataset[Tensor]) -> Tuple[Dataset[Tensor], Dataset[T
def create_trainer() -> Trainer:
config_max_epochs = CONFIG_PARSER.getint('training.trainer', 'max_epochs')
config_precision = CONFIG_PARSER.get('training.trainer', 'precision')
config_strategy = CONFIG_PARSER.get('training.trainer', 'strategy')
config_precision = CONFIG_PARSER.get('training.trainer', 'precision')
config_logger_path = CONFIG_PARSER.get('training.trainer', 'logger_path')
config_logger_name = CONFIG_PARSER.get('training.trainer', 'logger_name')
config_directory_path = CONFIG_PARSER.get('training.output', 'directory_path')
config_file_pattern = CONFIG_PARSER.get('training.output', 'file_pattern')
logger = TensorBoardLogger('.logs', name = 'face_swapper')
logger = TensorBoardLogger(config_logger_path, name = config_logger_name)
return Trainer(
logger = logger,
log_every_n_steps = 10,
max_epochs = config_max_epochs,
precision = config_precision,
strategy = config_strategy,
precision = config_precision,
callbacks =
[
ModelCheckpoint(