diff --git a/egs/librispeech/ASR/conformer_ctc/subsampling.py b/egs/librispeech/ASR/conformer_ctc/subsampling.py index d1ff7f233..d7be46f17 100644 --- a/egs/librispeech/ASR/conformer_ctc/subsampling.py +++ b/egs/librispeech/ASR/conformer_ctc/subsampling.py @@ -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 diff --git a/egs/librispeech/ASR/transducer_stateless/train.py b/egs/librispeech/ASR/transducer_stateless/train.py index d75341a07..80febc677 100755 --- a/egs/librispeech/ASR/transducer_stateless/train.py +++ b/egs/librispeech/ASR/transducer_stateless/train.py @@ -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