修复gpt训练卡死问题和unmatched '}' in format string问题

2 years ago · 59f35adad8
parent 57dde51e11
commit 59f35adad8
1 changed files with 1 additions and 0 deletions
--- a/GPT_SoVITS/s1_train.py
+++ b/GPT_SoVITS/s1_train.py
@ -106,6 +106,7 @@ def main(args):
        dirpath=ckpt_dir,
    )
    logger = TensorBoardLogger(name=output_dir.stem, save_dir=output_dir)
    os.environ["MASTER_ADDR"]="localhost"
    trainer: Trainer = Trainer(
        max_epochs=config["train"]["epochs"],
        accelerator="gpu",