HoneyTian commited on
Commit
360c8df
·
1 Parent(s): 59e7ec3
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