espnet / espnet/espnet_onnx

Model Export Error in S3PRL

Open
#122 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
169
Forks
25
PR merge metrics
No merged PRs in 30d

Description

I got an error trying to export my model into ONNX

```
Traceback (most recent call last):
File "convertmodel.py", line 4, in
m.export_from_zip(
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/export_asr.py", line 197, in export_from_zip
self.export(model, tag_name, quantize, optimize)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/export_asr.py", line 78, in export
enc_model = get_encoder(
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/__init__.py", line 60, in get_encoder
return DefaultEncoder(_model, frontend, **export_config)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/encoder_wrapper.py", line 18, in __init__
self.get_frontend(kwargs)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/encoder_wrapper.py", line 29, in get_frontend
self.frontend_model = get_frontend_models(self.frontend, kwargs)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/__init__.py", line 81, in get_frontend_models
return S3PRLModel(model, **export_config)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/frontends/s3prl.py", line 143, in __init__
self.upstream = HubertModel(model.upstream, max_seq_len)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/frontends/s3prl.py", line 77, in __init__
raise RuntimeError(
RuntimeError: S3PRL version is not supported. Please install v0.3 or v0.4
```
I downgraded my s3prl version from v0.4.17 -> v0.4.0 due to unsupported version error then i got this error after that.

```
Traceback (most recent call last):
File "convertmodel.py", line 4, in
m.export_from_zip(
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/export_asr.py", line 205, in export_from_zip
model = Speech2Text(**model_config)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/typeguard/__init__.py", line 1015, in wrapper
retval = func(*args, **kwargs)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet2/bin/asr_inference.py", line 147, in __init__
asr_model, asr_train_args = task.build_model_from_file(
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/typeguard/__init__.py", line 1015, in wrapper
retval = func(*args, **kwargs)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet2/tasks/abs_task.py", line 2329, in build_model_from_file
model = cls.build_model(args)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/typeguard/__init__.py", line 1015, in wrapper
retval = func(*args, **kwargs)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet2/tasks/asr.py", line 538, in build_model
frontend = frontend_class(**args.frontend_conf)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/typeguard/__init__.py", line 1015, in wrapper
retval = func(*args, **kwargs)
File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet2/asr/frontend/s3prl.py", line 47, in __init__
upstream = S3PRLUpstream(
TypeError: __init__() got an unexpected keyword argument 'normalize'

```
I tried commenting out that line and the lines after that since it still showed an error. Is this due to a broken installation? or version mismatch among frameworks?
I have Espnet v202503 and S3PRL v0.4.17 installed
Please add support for latest versions of Espnet and S3PRL. Thank you!

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.