Feat: support V220 train (maybe)

train_ms_V210.py

@@ -329,6 +329,16 @@ def run():
         if net_dur_disc is not None:
             scheduler_dur_disc.step()
 
+        if epoch == hps.train.epochs:
+            utils.save_compressed_models_checkpoint(
+                net_g,
+                epoch,
+                os.path.join(
+                    hps.model_dir,
+                    f"release_{global_step}.pth",
+                ),
+            )
+
 
 def train_and_evaluate(
     rank,
@@ -589,8 +599,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,