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

Rename to resume_path only

Browse Source
This commit is contained in:
henryruhs
2025-03-11 14:43:09 +01:00
parent c1bed34c27
commit 04eaa831ea
8 changed files with 11 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/training.py
+3 -3
View File
@@ -129,7 +129,7 @@ def train() -> None:
same_person_probability = CONFIG.getfloat('preparing.dataset', 'same_person_probability')
batch_size = CONFIG.getint('training.loader', 'batch_size')
num_workers = CONFIG.getint('training.loader', 'num_workers')
resume_file_path = CONFIG.get('training.output', 'resume_file_path')
output_resume_path = CONFIG.get('training.output', 'resume_path')
dataset = DataLoader(dataset_path, dataset_image_pattern, dataset_directory_pattern, same_person_probability)
training_loader = TorchDataLoader(dataset, batch_size = batch_size, shuffle = True, num_workers = num_workers, drop_last = True, pin_memory = True, persistent_workers = True)
@@ -137,7 +137,7 @@ def train() -> None:
face_swapper_trainer = FaceSwapperTrainer()
trainer = create_trainer()
if os.path.isfile(resume_file_path):
trainer.fit(face_swapper_trainer, training_loader, validation_loader, ckpt_path = resume_file_path)
if os.path.isfile(output_resume_path):
trainer.fit(face_swapper_trainer, training_loader, validation_loader, ckpt_path = output_resume_path)
else:
trainer.fit(face_swapper_trainer, training_loader, validation_loader)