Feat: support V220 train (maybe)

2023-12-23 15:18:27 +09:00
parent 2b66bde9a4
commit c2aabe15b7
11 changed files with 1319 additions and 7 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -740,8 +740,10 @@ def train_and_evaluate(
                        n_ckpts_to_keep=keep_ckpts,
                        sort_by_time=True,
                    )
-                save_compressed_models = hps.train.save_compressed_models
-                if save_compressed_models:
+                if (
+                    "save_compressed_models" in hps.train.keys()
+                    and hps.train.save_compressed_models is True
+                ):
                    utils.save_compressed_models_checkpoint(
                        net_g,
                        epoch,