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,
threshold: 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 channel_dim < 0:
channel_dim += x.ndim
@ -452,7 +452,7 @@ class DerivBalancer(torch.nn.Module):
def __init__(self, channel_dim: int,
threshold: float = 0.05,
max_factor: float = 0.02,
min_abs: float = 0.2):
min_abs: float = 0.5):
super(DerivBalancer, self).__init__()
self.channel_dim = channel_dim
self.threshold = threshold

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

@ -110,7 +110,7 @@ def get_parser():
parser.add_argument(
"--exp-dir",
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.
It specifies the directory where all training related
files, e.g., checkpoints, log, etc, are saved