未验证 提交 2fd904e9 编写于 作者: H Hui Zhang 提交者: GitHub

Merge pull request #1998 from KPatr1ck/resource

[Resource] Fix windows issue in paddlespeech.resource
......@@ -82,6 +82,7 @@ class CommonTaskResource:
self.model_tag = model_tag
self.version = version
self.res_dict = self.pretrained_models[model_tag][version]
self.format_path(self.res_dict)
self.res_dir = self._fetch(self.res_dict,
self._get_model_dir(model_type))
else:
......@@ -89,9 +90,19 @@ class CommonTaskResource:
self.voc_model_tag = model_tag
self.voc_version = version
self.voc_res_dict = self.pretrained_models[model_tag][version]
self.format_path(self.voc_res_dict)
self.voc_res_dir = self._fetch(self.voc_res_dict,
self._get_model_dir(model_type))
@staticmethod
def format_path(res_dict: Dict[str, str]):
for k, v in res_dict.items():
if '/' in v:
if v.startswith('https://') or v.startswith('http://'):
continue
else:
res_dict[k] = os.path.join(*(v.split('/')))
@staticmethod
def get_model_class(model_name) -> List[object]:
"""Dynamic import model class.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册