From d5784e3444ff891b92c681d866f1d527a25cb299 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期日, 23 四月 2023 15:51:59 +0800
Subject: [PATCH] Merge pull request #404 from alibaba-damo-academy/main

---
 egs/aishell/transformer/utils/prepare_checkpoint.py |   18 ++++++++++++++++++
 1 files changed, 18 insertions(+), 0 deletions(-)

diff --git a/egs/aishell/transformer/utils/prepare_checkpoint.py b/egs/aishell/transformer/utils/prepare_checkpoint.py
new file mode 100644
index 0000000..4d5e8a0
--- /dev/null
+++ b/egs/aishell/transformer/utils/prepare_checkpoint.py
@@ -0,0 +1,18 @@
+import os
+import shutil
+
+from modelscope.hub.snapshot_download import snapshot_download
+
+
+if __name__ == '__main__':
+    import sys
+    
+    model = sys.argv[1]
+    checkpoint_dir = sys.argv[2]
+    checkpoint_name = sys.argv[3]
+    
+    try:
+        pretrained_model_path = snapshot_download(model, cache_dir=checkpoint_dir)
+    except BaseException:
+        raise BaseException(f"Please download pretrain model from ModelScope firstly.")
+    shutil.copy(os.path.join(checkpoint_dir, checkpoint_name), os.path.join(pretrained_model_path, "model.pb"))

--
Gitblit v1.9.1