AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
112 Commits 2 Branches 0 Tags
Commit Graph

112 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Binbin Zhang
dc1ac8fecd
[examples] use big model for ds_tcn (#47) 2021-12-15 11:12:24 +08:00
Binbin Zhang
e3bfcf9f4e
[doc] add quantize result (#46) 2021-12-15 11:07:59 +08:00
Binbin Zhang
f86a797b10
[kws] add static quantize (#44) 
* [kws] add static quantize

* refine lint error in shuffle_list.py

* refine lint

* fix topo
 2021-12-14 14:32:54 +08:00
ZY
05bb5d1bdb
[doc] update readme (#45) 2021-12-14 09:38:57 +08:00
Binbin Zhang
a61db05ff4
[examples] add weight decay in aishell (#43) 2021-12-13 19:53:24 +08:00
Binbin Zhang
171309bd9e
[bin] use torchrun to launch ddp training (#42) 2021-12-13 19:45:55 +08:00
Binbin Zhang
28652a766f
[kws] support fuse and quantization (#41) 2021-12-13 19:16:50 +08:00
Binbin Zhang
fd255fd7c6
[examples] update spec aug parameters in hi xiaowen(#40) 2021-12-10 15:48:16 +08:00
Binbin Zhang
bc8d9f1c37
[examples] fix mdtc small config in hi_xiaowen (#39) 2021-12-09 17:53:34 +08:00
Menglong Xu
4a875776e5
[example] support hey_snips_kws_4.0 dataset (#38) 
* [example] support hey_snips_kws_4.0 dataset

* format

* format
 2021-12-08 23:46:05 +08:00
Menglong Xu
1eda27647b
[fix bug] add sigmoid() in score.py (#37) 
The sigmoid() in kws/model/kws_model.py:KWSModel() was moved into kws/model/loss.py:max_pooling_loss()
To compute the posterior score correctly, the sigmoid() should also be added to kws/bin/score.py:main()
 2021-12-08 13:41:20 +08:00
xiaohou
bd504c3cee
[bin] check torch script (#36) 
* update run.sh

* [network] add code to check whether the model can be exported to torch script or not
 2021-12-07 11:17:47 +08:00
xiaohou
afbc1d2960
[example] add testing code for speech command dataset (#32) 
* update run.sh

* update run.sh

* rename test.py to compute_accuracy.py

* update run,sh
 2021-12-07 10:56:30 +08:00
Binbin Zhang
b55ae111ae
[model] refactor tcn and ds_tcn share the same base class (#35) 2021-12-07 10:52:14 +08:00
Binbin Zhang
92a4c19ffe
[examples] use ds_tcn as default model (#34) 
* [examples] use ds_tcn as default model

* fix scoring gpu id
 2021-12-07 10:36:38 +08:00
Binbin Zhang
93bf93565e
[doc] add wechat group discuss (#33) 2021-12-06 22:00:54 +08:00
Binbin Zhang
c7c5bd3edc
[kws] refine tcn and ds_tcp, add batchnorm (#31) 
* [kws] fix seed type

* [kws] refine tcn and ds_tcn, add batch norm
 2021-12-06 17:24:48 +08:00
xiaohou
37f56db5af
[exampels] add speechcommand train (#30) 
* [example] added code for training speech command dataset

* update kes_model.py

* update kes_model.py

* format

* format

* add more comments to explain the new classifier designed for speech command classification task

* add copyrigh info

* update copyrigh info of classifier.py
 2021-12-06 17:14:33 +08:00
xiaohou
8be4bef405
[examples] speech command data prepare (#27) 
* [examples] added speech command data preparation code

* update

* updata path.sh
 2021-12-06 12:00:25 +08:00
Binbin Zhang
5241491e95
[kws] fix seed type (#26) 2021-12-06 11:38:41 +08:00
Binbin Zhang
a5a54782cc
[kws] use **kvargs for optim to reduce code (#25) 2021-12-06 11:23:44 +08:00
Menglong Xu
88444ab177
[examples] correct a spelling mistake (#24) 2021-12-05 21:24:06 +08:00
Binbin Zhang
8cfd4ed4f2
[kws] fix weight_decay key error (#23) 2021-12-04 17:19:46 +08:00
Binbin Zhang
dfe8b2536b
Revert "[recipe] suport speech command dataset (#21)" (#22) 
This reverts commit c48c959807e7e80cdd514be9bd019b16e3b816eb.
 2021-12-04 13:55:58 +08:00
xiaohou
c48c959807
[recipe] suport speech command dataset (#21) 
* [recipe] suport speech command dataset

* format

* format

* format

* update run.sh
 2021-12-03 21:07:42 +08:00
Binbin Zhang
6416f9c7db
Merge pull request #20 from wenet-e2e/fix-jingyonghou 
fix bug in checking kernel size, kernel size should be a odd number
 2021-12-03 10:29:27 +08:00
jingyong hou
8ea11dc572 fix bug in checking kernel size, kernel size should be a odd number 2021-12-03 10:25:02 +08:00
Binbin Zhang
56bdce8775
[doc] add reference (#18) 
* [doc] add reference

* [doc] refine format

* [doc] refine format

* fix url
 2021-11-30 20:56:26 +08:00
Binbin Zhang
1909fcd360
Merge pull request #17 from wenet-e2e/dev_jingyonghou 
[network] update mdtc.py to prevent possible errors and remove useless functions
 2021-11-30 17:20:09 +08:00
xiaohou
5236d42800
Update mdtc.py 2021-11-30 17:14:30 +08:00
xiaohou
b9551bb716
Update mdtc.py 2021-11-30 17:07:01 +08:00
xiaohou
8a50252c64
Update mdtc.py 2021-11-30 17:05:48 +08:00
Jingyong Hou
f642c0952a update mdtc.py to prevent possible errors and remove useless functions 2021-11-30 16:47:35 +08:00
lxiao336
ba6919baaf
modifications to get the mdtc model torch-scriptable (#14) 
* modifying some implmentations of mdtc to get the model torch-scripting through

* modifications to get the mdtc model torch-scriptable

Co-authored-by: lxiao336 <shawl336@163.com>
 2021-11-29 11:15:30 +08:00
Binbin Zhang
bf3029c0fa
[doc] rename to wekws (#15) 2021-11-28 21:50:50 +08:00
Binbin Zhang
5bc7c8d64e
Merge pull request #12 from wenet-e2e/dev-jingyonghou 
[fix bug] resolve bugs in score.py
 2021-11-22 09:33:42 +08:00
jingyong hou
a681a8c931 [fix bug] resolve bugs in score.py 2021-11-21 23:20:16 +08:00
Binbin Zhang
87c42add2a
Merge pull request #11 from wenet-e2e/jingyong-random-seed 
add mannul random seed so we can reproduce the experimental results
 2021-11-19 17:41:31 +08:00
jingyong hou
17cd3c47e5 update train.py 2021-11-19 17:18:58 +08:00
jingyong hou
8cd12edfed update loss.py and kws_model.py 2021-11-19 17:14:11 +08:00
Binbin Zhang
f7fd62db7d
Merge pull request #9 from wenet-e2e/dev-jingyonghou 
add results of mdtc
 2021-11-19 17:01:23 +08:00
jingyong hou
9aaa4fc26c add mannul random seed so we can reproduce the experimental results 2021-11-19 16:23:03 +08:00
jingyong hou
edfc6de743 add results of mdtc 2021-11-19 15:31:11 +08:00
Binbin Zhang
2072abeb70
[doc] add install guide (#8) 2021-11-19 12:24:47 +08:00
Binbin Zhang
3dde87bdcd
[github] fix lint rule (#7) 2021-11-19 11:08:55 +08:00
xiaohou
1d5f42f0a3
Merge pull request #4 from wenet-e2e/binbin-docs 
[docs] support docs
 2021-11-11 10:08:08 +08:00
Binbin Zhang
0e852bdcbb [docs] support docs 2021-11-11 10:03:52 +08:00
xiaohou
531d795bce
Merge pull request #3 from jingyonghou/master 
add MDTC model support
 2021-11-11 10:01:03 +08:00
Jingyong Hou
9514336cc4 formatting 2021-11-11 09:57:35 +08:00
Jingyong Hou
603662f152 add a simple intro to MDTC 2021-11-11 09:55:29 +08:00