blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-06 03:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update s1_train.py

Browse Source
This commit is contained in:
RVC-Boss 2024-06-29 22:54:05 +08:00 committed by GitHub
parent 1b4c01e860
commit a208698e77
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
GPT_SoVITS/s1_train.py
View File

@ -79,15 +79,17 @@ class my_model_ckpt(ModelCheckpoint):
to_save_od["config"] = self.config to_save_od["config"] = self.config
to_save_od["info"] = "GPT-e%s" % (trainer.current_epoch + 1) to_save_od["info"] = "GPT-e%s" % (trainer.current_epoch + 1)
# torch.save( # torch.save(
my_save( # print(os.environ)
to_save_od, if(os.environ.get("LOCAL_RANK","0")=="0"):
"%s/%s-e%s.ckpt" my_save(
% ( to_save_od,
self.half_weights_save_dir, "%s/%s-e%s.ckpt"
self.exp_name, % (
trainer.current_epoch + 1, self.half_weights_save_dir,
), self.exp_name,
) trainer.current_epoch + 1,
),
)
self._save_last_checkpoint(trainer, monitor_candidates) self._save_last_checkpoint(trainer, monitor_candidates)