update
Browse files
toolbox/torchaudio/models/vad/fsmn_vad/configuration_fsmn_vad.py
CHANGED
@@ -27,6 +27,9 @@ class FSMNVadConfig(PretrainedConfig):
|
|
27 |
|
28 |
use_softmax: bool = False,
|
29 |
|
|
|
|
|
|
|
30 |
lr: float = 0.001,
|
31 |
lr_scheduler: str = "CosineAnnealingLR",
|
32 |
lr_scheduler_kwargs: dict = None,
|
@@ -64,6 +67,10 @@ class FSMNVadConfig(PretrainedConfig):
|
|
64 |
|
65 |
self.use_softmax = use_softmax
|
66 |
|
|
|
|
|
|
|
|
|
67 |
# train
|
68 |
self.lr = lr
|
69 |
self.lr_scheduler = lr_scheduler
|
|
|
27 |
|
28 |
use_softmax: bool = False,
|
29 |
|
30 |
+
min_snr_db: float = -10,
|
31 |
+
max_snr_db: float = 20,
|
32 |
+
|
33 |
lr: float = 0.001,
|
34 |
lr_scheduler: str = "CosineAnnealingLR",
|
35 |
lr_scheduler_kwargs: dict = None,
|
|
|
67 |
|
68 |
self.use_softmax = use_softmax
|
69 |
|
70 |
+
# data snr
|
71 |
+
self.min_snr_db = min_snr_db
|
72 |
+
self.max_snr_db = max_snr_db
|
73 |
+
|
74 |
# train
|
75 |
self.lr = lr
|
76 |
self.lr_scheduler = lr_scheduler
|