Binbin Zhang
57021924cb
[kws] support onnx export ( #53 )
2022-01-15 13:50:34 +08:00
Menglong Xu
f622c55b04
[kws] update parameter for plotting det curve ( #54 )
2021-12-17 20:52:45 +08:00
Menglong Xu
768900307a
[kws] add code for plotting det curve ( #52 )
...
* [kws] add code for plotting det curve
* format
* format
* format
* format
* [kws] add code for plotting det curve
format
format
format
format
* set xlim and ylim by parameter
* set xlim and ylim optional
* update help information
* update parser type
* Update run.sh
2021-12-16 18:21:04 +08:00
Binbin Zhang
8943acb51f
[kws] put activation in model, so the activation could be exported in script model
2021-12-15 21:10:27 +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
Binbin Zhang
171309bd9e
[bin] use torchrun to launch ddp training ( #42 )
2021-12-13 19:45:55 +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
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
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
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
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
jingyong hou
17cd3c47e5
update train.py
2021-11-19 17:18:58 +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
3326c6d37f
formatting
2021-11-11 09:30:37 +08:00
Jingyong Hou
0942092426
format the code
2021-11-10 22:49:53 +08:00
Jingyong Hou
7df9ced666
fixed bug of compute_cmvn_stats.py
2021-11-10 22:40:21 +08:00
Jingyong Hou
4db050eb67
add model mdtc for mobvoi-hotword example
2021-11-10 22:13:46 +08:00
Binbin Zhang
aa0b0c11a8
[kws] add kws base code
2021-11-10 18:48:57 +08:00
