Skip to content

Commit 7ae9d17

Browse files
committed
fix incorrect flag in finetuning example
1 parent 9ce4837 commit 7ae9d17

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

TrainingInterfaces/TrainingPipelines/finetuning_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def run(gpu_id, resume_checkpoint, finetune, model_dir, resume, use_wandb, wandb
8080
# DOWNLOAD THESE INITIALIZATION MODELS FROM THE RELEASE PAGE OF THE GITHUB OR RUN THE DOWNLOADER SCRIPT TO GET THEM AUTOMATICALLY
8181
path_to_checkpoint=os.path.join(MODELS_DIR, "ToucanTTS_Meta", "best.pt") if resume_checkpoint is None else resume_checkpoint,
8282
path_to_embed_model=os.path.join(MODELS_DIR, "Embedding", "embedding_function.pt"),
83-
fine_tune=True if resume_checkpoint is None else finetune,
83+
fine_tune=True if resume_checkpoint is None and not resume else finetune,
8484
resume=resume,
8585
steps=5000,
8686
use_wandb=use_wandb)

0 commit comments

Comments
 (0)