From e84f17adca2d8a81bc2d0229b9531e7eb0a7705c Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 26 三月 2024 12:34:26 +0800
Subject: [PATCH] update
---
funasr/register.py | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/funasr/register.py b/funasr/register.py
index ebfdaac..45e2a85 100644
--- a/funasr/register.py
+++ b/funasr/register.py
@@ -15,6 +15,7 @@
predictor_classes = {}
stride_conv_classes = {}
tokenizer_classes = {}
+ dataloader_classes = {}
batch_sampler_classes = {}
dataset_classes = {}
index_ds_classes = {}
@@ -29,7 +30,7 @@
flag = key in classes_key
if classes_key.endswith("_meta") and flag:
print(f"----------- ** {classes_key.replace('_meta', '')} ** --------------")
- headers = ["class name", "class location"]
+ headers = ["register name", "class name", "class location"]
metas = []
for register_key, meta in classes_dict.items():
metas.append(meta)
@@ -67,8 +68,8 @@
class_line = inspect.getsourcelines(target_class)[1]
pattern = r'^.+/funasr/'
class_file = re.sub(pattern, 'funasr/', class_file)
- meata_data = [f"{target_class.__name__}", f"{class_file}:{class_line}"]
- # meata_data = [f"{target_class.__name__}", f"{registry_key}", f"{class_file}:{class_line}"]
+ # meata_data = [f"{target_class.__name__}", f"{class_file}:{class_line}"]
+ meata_data = [f"{registry_key}", f"{target_class.__name__}", f"{class_file}:{class_line}"]
registry_meta[registry_key] = meata_data
# print(f"Registering class: {class_file}:{class_line} - {target_class.__name__} as {registry_key}")
return target_class
--
Gitblit v1.9.1