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

修复gpt训练卡死问题和unmatched '}' in format string问题
This commit is contained in:
RVC-Boss 2024-02-08 21:53:31 +08:00 committed by GitHub
parent 57dde51e11
commit 59f35adad8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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",