Comments (2)
I performed ONNX check procedure. It seems that it is OK. Please that output of ONNX check below.
(base) root@gpu-alex-0-0:/workspace/inputs/alexg/asr/src/models/k2_2024/icefall/egs/tedlium3/ASR# sh/main_onnx_check.sh
stage 2
2024-05-20 19:36:12,134 INFO [onnx_check.py:201] {'jit_filename': 'zipformer/exp/jit_script.pt', 'onnx_encoder_filename': 'zipformer/exp/encoder-epoch-50-avg-1.onnx', 'onnx_decoder_filename': 'zipformer/exp/decoder-epoch-50-avg-1.onnx', 'onnx_joiner_filename': 'zipformer/exp/joiner-epoch-50-avg-1.onnx'}
2024-05-20 18:36:13.453391447 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/0/encoder_pos/Shape_5_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453442414 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/0/encoder_pos/Constant_46_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453451091 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/0/encoder_pos/Constant_24_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453459306 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/0/encoder_pos/Constant_48_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453467442 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/0/encoder_pos/Shape_10_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453475507 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '927'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453482620 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/0/encoder_pos/Constant_22_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453488101 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/0/encoder_pos/Shape_15_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453494102 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/0/encoder_pos/Constant_37_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453500725 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/0/encoder_pos/Constant_39_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453644267 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/1/encoder/encoder_pos/Shape_5_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453651361 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/1/encoder/encoder_pos/Constant_46_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453660819 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/1/encoder/encoder_pos/Shape_10_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453667391 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/1/encoder/encoder_pos/Constant_22_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453674895 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '2181'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453682550 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/1/encoder/encoder_pos/Constant_24_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453691808 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/1/encoder/encoder_pos/Constant_48_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453698701 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/1/encoder/encoder_pos/Shape_15_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453706966 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/1/encoder/encoder_pos/Constant_37_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453713559 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/1/encoder/encoder_pos/Constant_39_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453827274 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/2/encoder/encoder_pos/Shape_15_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453834799 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/2/encoder/encoder_pos/Shape_10_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453841492 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/2/encoder/encoder_pos/Shape_5_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453848164 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/2/encoder/encoder_pos/Constant_48_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453855137 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/2/encoder/encoder_pos/Constant_46_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453861900 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/2/encoder/encoder_pos/Constant_24_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453868803 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/2/encoder/encoder_pos/Constant_37_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453877590 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '3485'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453887208 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/2/encoder/encoder_pos/Constant_22_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.453894442 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/2/encoder/encoder_pos/Constant_39_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454000173 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/3/encoder/encoder_pos/Shape_15_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454007797 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/3/encoder/encoder_pos/Constant_48_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454014870 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/3/encoder/encoder_pos/Constant_24_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454021794 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/3/encoder/encoder_pos/Constant_39_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454028827 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/3/encoder/encoder_pos/Constant_46_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454036602 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/3/encoder/encoder_pos/Constant_22_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454043094 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/3/encoder/encoder_pos/Shape_5_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454051670 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '5253'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454059174 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/3/encoder/encoder_pos/Shape_10_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454067721 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/3/encoder/encoder_pos/Constant_37_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454173111 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/4/encoder/encoder_pos/Shape_15_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454180384 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/4/encoder/encoder_pos/Shape_5_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454187568 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/4/encoder/encoder_pos/Constant_48_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454196926 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/4/encoder/encoder_pos/Shape_10_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454203749 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/4/encoder/encoder_pos/Constant_22_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454211023 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '7465'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454217365 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/4/encoder/encoder_pos/Constant_24_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454225570 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/4/encoder/encoder_pos/Constant_37_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454232053 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/4/encoder/encoder_pos/Constant_46_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454239026 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/4/encoder/encoder_pos/Constant_39_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454330389 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/5/encoder/encoder_pos/Shape_10_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454337062 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/5/encoder/encoder_pos/Shape_5_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454343554 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/5/encoder/encoder_pos/Constant_48_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454349505 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/5/encoder/encoder_pos/Constant_46_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454356258 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/5/encoder/encoder_pos/Shape_15_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454362490 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/5/encoder/encoder_pos/Constant_22_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454367910 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/5/encoder/encoder_pos/Constant_24_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454375495 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/5/encoder/encoder_pos/Constant_37_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454381356 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '/encoder/5/encoder/encoder_pos/Constant_39_output_0'. It is not used by any node and should be removed from the model.
2024-05-20 18:36:13.454387718 [W:onnxruntime:, graph.cc:3211 CleanUnusedInitializers] Removing initializer '9211'. It is not used by any node and should be removed from the model.
2024-05-20 19:36:15,327 INFO [onnx_pretrained.py:168] context_size: 2
2024-05-20 19:36:15,327 INFO [onnx_pretrained.py:169] vocab_size: 500
2024-05-20 19:36:15,331 INFO [onnx_pretrained.py:181] joiner_dim: 512
2024-05-20 19:36:15,331 INFO [onnx_check.py:211] Test encoder
2024-05-20 19:36:15,332 INFO [onnx_check.py:132] test_encoder: iter 0, N=9, T=32
2024-05-20 19:36:16,229 INFO [onnx_check.py:132] test_encoder: iter 1, N=16, T=47
2024-05-20 19:36:17,357 INFO [onnx_check.py:132] test_encoder: iter 2, N=13, T=40
2024-05-20 19:36:18,157 INFO [onnx_check.py:214] Test decoder
2024-05-20 19:36:18,157 INFO [onnx_check.py:156] test_decoder: iter 0, N=12
2024-05-20 19:36:18,164 INFO [onnx_check.py:156] test_decoder: iter 1, N=62
2024-05-20 19:36:18,171 INFO [onnx_check.py:156] test_decoder: iter 2, N=75
2024-05-20 19:36:18,180 INFO [onnx_check.py:156] test_decoder: iter 3, N=82
2024-05-20 19:36:18,189 INFO [onnx_check.py:156] test_decoder: iter 4, N=56
2024-05-20 19:36:18,195 INFO [onnx_check.py:156] test_decoder: iter 5, N=52
2024-05-20 19:36:18,201 INFO [onnx_check.py:156] test_decoder: iter 6, N=39
2024-05-20 19:36:18,205 INFO [onnx_check.py:156] test_decoder: iter 7, N=26
2024-05-20 19:36:18,209 INFO [onnx_check.py:156] test_decoder: iter 8, N=80
2024-05-20 19:36:18,217 INFO [onnx_check.py:156] test_decoder: iter 9, N=19
2024-05-20 19:36:18,220 INFO [onnx_check.py:217] Test joiner
2024-05-20 19:36:18,220 INFO [onnx_check.py:181] test_joiner: iter 0, N=66
2024-05-20 19:36:18,232 INFO [onnx_check.py:181] test_joiner: iter 1, N=29
2024-05-20 19:36:18,237 INFO [onnx_check.py:181] test_joiner: iter 2, N=89
2024-05-20 19:36:18,253 INFO [onnx_check.py:181] test_joiner: iter 3, N=17
2024-05-20 19:36:18,256 INFO [onnx_check.py:181] test_joiner: iter 4, N=86
2024-05-20 19:36:18,268 INFO [onnx_check.py:181] test_joiner: iter 5, N=62
2024-05-20 19:36:18,278 INFO [onnx_check.py:181] test_joiner: iter 6, N=82
2024-05-20 19:36:18,290 INFO [onnx_check.py:181] test_joiner: iter 7, N=42
2024-05-20 19:36:18,296 INFO [onnx_check.py:181] test_joiner: iter 8, N=92
2024-05-20 19:36:18,309 INFO [onnx_check.py:181] test_joiner: iter 9, N=88
2024-05-20 19:36:18,321 INFO [onnx_check.py:219] Finished checking ONNX models
from icefall.
The solution is to work with wav duration less than 20 sec.
from icefall.
Related Issues (20)
- pytorch ver. `>=2.1.0` breaks compatibility with all `conformer_ctc` recipes
- Multi Lingual model HOT 1
- low resource data HOT 1
- Identical Batches Across Multiple GPUs HOT 2
- CTC/AED PROBLEM IN K2 HOT 10
- append features HOT 1
- CTC/AED PROBLEMS IN EXPORTING JIT MODULE HOT 4
- Error happens with egs/librispeech/ASR/prepare_mmi.sh HOT 3
- Use CutSet.mux to effect? HOT 10
- Help with training/finetuning a zipformer based model HOT 6
- Different Training Loss with Single Node (8 GPUs) vs. Two Nodes (4 GPUs Each)
- Data cleaning HOT 3
- OTC with conformer librispeech/WASR isn't converage.
- ONNX bug HOT 9
- Questions about modifying prepare.sh for training ASR model on custom data HOT 2
- How to use my own dataset based on another dataset HOT 3
- kaldifeat installation error HOT 2
- Why unique lexicon is needed in Chinese ASR, but not in English ASR?
- Error during training OTC conformer_ctc2 HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from icefall.