Update s1_train.py

1 year ago · a208698e77
parent 1b4c01e860
commit a208698e77
1 changed files with 11 additions and 9 deletions
--- a/GPT_SoVITS/s1_train.py
+++ b/GPT_SoVITS/s1_train.py
@ -79,6 +79,8 @@ class my_model_ckpt(ModelCheckpoint):
                    to_save_od["config"] = self.config
                    to_save_od["info"] = "GPT-e%s" % (trainer.current_epoch + 1)
                    # torch.save(
+                    # print(os.environ)
+                    if(os.environ.get("LOCAL_RANK","0")=="0"):
                        my_save(
                            to_save_od,
                            "%s/%s-e%s.ckpt"