change naming logic of saving trained file for synthesizer to allow shorter interval

3 years ago · 8ad9ba2b60
parent b56ec5ee1b
commit 8ad9ba2b60
1 changed files with 1 additions and 1 deletions
--- a/synthesizer/train.py
+++ b/synthesizer/train.py
@ -222,7 +222,7 @@ def train(run_id: str, syn_dir: str, models_dir: str, save_every: int,

                # Backup or save model as appropriate
                if backup_every != 0 and step % backup_every == 0 : 
-                    backup_fpath = Path("{}/{}_{}k.pt".format(str(weights_fpath.parent), run_id, k))
+                    backup_fpath = Path("{}/{}_{}.pt".format(str(weights_fpath.parent), run_id, step))
                    model.save(backup_fpath, optimizer)

                if save_every != 0 and step % save_every == 0 :