fix model dir

Mddct · Sep 5, 2024 · 05f109e · 05f109e
1 parent 017d27f
commit 05f109e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wenet/codec/vocos_low_latency/train.py b/wenet/codec/vocos_low_latency/train.py
@@ -410,7 +410,7 @@ def main():
 
         # save interval
         if i % 2000 == 0:
-            save_checkpoint(args.model_path, train_state)
+            save_checkpoint(args.model_dir, train_state)
 
 
 if __name__ == '__main__':