mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-05-09 22:29:06 +08:00
修复gpt训练卡死问题和unmatched '}' in format string问题
修复gpt训练卡死问题和unmatched '}' in format string问题
This commit is contained in:
parent
92366718cc
commit
12fd65f67d
@ -106,6 +106,7 @@ def main(args):
|
|||||||
dirpath=ckpt_dir,
|
dirpath=ckpt_dir,
|
||||||
)
|
)
|
||||||
logger = TensorBoardLogger(name=output_dir.stem, save_dir=output_dir)
|
logger = TensorBoardLogger(name=output_dir.stem, save_dir=output_dir)
|
||||||
|
os.environ["MASTER_ADDR"]="localhost"
|
||||||
trainer: Trainer = Trainer(
|
trainer: Trainer = Trainer(
|
||||||
max_epochs=config["train"]["epochs"],
|
max_epochs=config["train"]["epochs"],
|
||||||
accelerator="gpu",
|
accelerator="gpu",
|
||||||
|
Loading…
x
Reference in New Issue
Block a user