[egs] Fix model saving path in DeepClustering recipes(asteroid-team#398)

* Fixes final checkpoint save for wsj0-mix DeepClustering Co-authored-by: Manuel Pariente <pariente.mnl@gmail.com>
sreyareddy15 · Feb 6, 2021 · 6ca575b · 6ca575b
1 parent 47e5237
commit 6ca575b
Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/egs/kinect-wsj/DeepClustering/train.py b/egs/kinect-wsj/DeepClustering/train.py
@@ -78,7 +78,7 @@ def main(conf):
     with open(os.path.join(exp_dir, "best_k_models.json"), "w") as f:
         json.dump(checkpoint.best_k_models, f, indent=0)
     # Save last model for convenience
-    torch.save(system.model.state_dict(), os.path.join(exp_dir, "checkpoints/final.pth"))
+    torch.save(system.model.state_dict(), os.path.join(exp_dir, "final_model.pth"))
 
 
 # TODO:Should ideally be inherited from wsj0-mix

diff --git a/egs/wsj0-mix/DeepClustering/train.py b/egs/wsj0-mix/DeepClustering/train.py
@@ -80,7 +80,7 @@ def main(conf):
     with open(os.path.join(exp_dir, "best_k_models.json"), "w") as f:
         json.dump(best_k, f, indent=0)
     # Save last model for convenience
-    torch.save(system.model.state_dict(), os.path.join(exp_dir, "checkpoints/final.pth"))
+    torch.save(system.model.state_dict(), os.path.join(exp_dir, "final_model.pth"))
 
 
 class ChimeraSystem(System):