From 8e740e9897834c77e2475790741b330c18e518dc Mon Sep 17 00:00:00 2001
From: season <season4675@gmail.com>
Date: 星期日, 02 七月 2023 19:38:48 +0800
Subject: [PATCH] [funasr-runtime-deploy 0.0.4] streamline the process, and add command line parameters. (#697)
---
funasr/models/frontend/s3prl.py | 6 ++----
1 files changed, 2 insertions(+), 4 deletions(-)
diff --git a/funasr/models/frontend/s3prl.py b/funasr/models/frontend/s3prl.py
index c0a526f..fdeb1c5 100644
--- a/funasr/models/frontend/s3prl.py
+++ b/funasr/models/frontend/s3prl.py
@@ -8,8 +8,8 @@
import humanfriendly
import torch
-from typeguard import check_argument_types
+from funasr.models.frontend.abs_frontend import AbsFrontend
from funasr.modules.frontends.frontend import Frontend
from funasr.modules.nets_utils import pad_list
from funasr.utils.get_default_kwargs import get_default_kwargs
@@ -26,7 +26,7 @@
return args
-class S3prlFrontend(torch.nn.Module):
+class S3prlFrontend(AbsFrontend):
"""Speech Pretrained Representation frontend structure for ASR."""
def __init__(
@@ -36,7 +36,6 @@
download_dir: str = None,
multilayer_feature: bool = False,
):
- assert check_argument_types()
super().__init__()
if isinstance(fs, str):
fs = humanfriendly.parse_size(fs)
@@ -99,7 +98,6 @@
def _tile_representations(self, feature):
"""Tile up the representations by `tile_factor`.
-
Input - sequence of representations
shape: (batch_size, seq_len, feature_dim)
Output - sequence of tiled representations
--
Gitblit v1.9.1