wav2vec2 demo update: support different optimizer and lr_schedular, align mdoel, update input type, test=asr

pull/2658/head
tianhao zhang 3 years ago
parent 9d56d8d3b2
commit fc62abf279

@ -22,7 +22,9 @@ from ..utils.dynamic_import import dynamic_import
from ..utils.env import MODEL_HOME from ..utils.env import MODEL_HOME
from .model_alias import model_alias from .model_alias import model_alias
task_supported = ['asr', 'cls', 'st', 'text', 'tts', 'vector', 'kws', 'ssl', 'whisper'] task_supported = [
'asr', 'cls', 'st', 'text', 'tts', 'vector', 'kws', 'ssl', 'whisper'
]
model_format_supported = ['dynamic', 'static', 'onnx'] model_format_supported = ['dynamic', 'static', 'onnx']
inference_mode_supported = ['online', 'offline'] inference_mode_supported = ['online', 'offline']

Loading…
Cancel
Save