Fixes final checkpoint save for wsj0-mix DeepClustering example.

asteroid-team · Dec 28, 2020 · 4b0068d · 4b0068d
1 parent 1dc1c13
commit 4b0068d
Showing 1 changed file with 1 addition and 1 deletion.
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, "best_model.pth"))
 
 
 class ChimeraSystem(System):