From c568628130ac42ebeea8cf48fe926520a31ff511 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 16 五月 2023 10:57:21 +0800
Subject: [PATCH] update repo
---
egs/aishell2/transformer/utils/proc_conf_oss.py | 35 +++++++++++++++++++++++++++++++++++
1 files changed, 35 insertions(+), 0 deletions(-)
diff --git a/egs/aishell2/transformer/utils/proc_conf_oss.py b/egs/aishell2/transformer/utils/proc_conf_oss.py
new file mode 100755
index 0000000..c4a90c5
--- /dev/null
+++ b/egs/aishell2/transformer/utils/proc_conf_oss.py
@@ -0,0 +1,35 @@
+from pathlib import Path
+
+import torch
+import yaml
+
+
+class NoAliasSafeDumper(yaml.SafeDumper):
+ # Disable anchor/alias in yaml because looks ugly
+ def ignore_aliases(self, data):
+ return True
+
+
+def yaml_no_alias_safe_dump(data, stream=None, **kwargs):
+ """Safe-dump in yaml with no anchor/alias"""
+ return yaml.dump(
+ data, stream, allow_unicode=True, Dumper=NoAliasSafeDumper, **kwargs
+ )
+
+
+def gen_conf(file, out_dir):
+ conf = torch.load(file)["config"]
+ conf["oss_bucket"] = "null"
+ print(conf)
+ output_dir = Path(out_dir)
+ output_dir.mkdir(parents=True, exist_ok=True)
+ with (output_dir / "config.yaml").open("w", encoding="utf-8") as f:
+ yaml_no_alias_safe_dump(conf, f, indent=4, sort_keys=False)
+
+
+if __name__ == "__main__":
+ import sys
+
+ in_f = sys.argv[1]
+ out_f = sys.argv[2]
+ gen_conf(in_f, out_f)
--
Gitblit v1.9.1