mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 10:02:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change min-abs threshold from 0.2 to 0.5

Browse Source
This commit is contained in:
Daniel Povey 2022-03-11 14:16:33 +08:00
parent bfce5f63e4
commit e3e14cf7a4
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/conformer_ctc/subsampling.py
View File

@ -316,7 +316,7 @@ class DerivBalancerFunction(torch.autograd.Function):
channel_dim: int, channel_dim: int,
threshold: float = 0.05, threshold: float = 0.05,
max_factor: float = 0.05, max_factor: float = 0.05,
min_abs: float = 0.2) -> Tensor: min_abs: float = 0.5) -> Tensor:
if x.requires_grad: if x.requires_grad:
if channel_dim < 0: if channel_dim < 0:
channel_dim += x.ndim channel_dim += x.ndim
@ -452,7 +452,7 @@ class DerivBalancer(torch.nn.Module):
def __init__(self, channel_dim: int, def __init__(self, channel_dim: int,
threshold: float = 0.05, threshold: float = 0.05,
max_factor: float = 0.02, max_factor: float = 0.02,
min_abs: float = 0.2): min_abs: float = 0.5):
super(DerivBalancer, self).__init__() super(DerivBalancer, self).__init__()
self.channel_dim = channel_dim self.channel_dim = channel_dim
self.threshold = threshold self.threshold = threshold

2
egs/librispeech/ASR/transducer_stateless/train.py
View File

@ -110,7 +110,7 @@ def get_parser():
parser.add_argument( parser.add_argument(
"--exp-dir", "--exp-dir",
type=str, type=str,
default="transducer_stateless/specaugmod_baseline_randcombine1_expscale3_brelu2swish2_0.1_bnorm2ma0.2", default="transducer_stateless/specaugmod_baseline_randcombine1_expscale3_brelu2swish2_0.1_bnorm2ma0.5",
help="""The experiment dir. help="""The experiment dir.
It specifies the directory where all training related It specifies the directory where all training related
files, e.g., checkpoints, log, etc, are saved files, e.g., checkpoints, log, etc, are saved