Revert "Reduce const_attention_rate"

This reverts commit bc002a9eda8ac912cff235460dcdef2fd51b2f19.
This commit is contained in:
Daniel Povey 2022-12-17 13:27:17 +08:00
parent b9326e1ef2
commit 35b63c1387

View File

@ -394,7 +394,7 @@ class ZipformerEncoderLayer(nn.Module):
layer_skip_rate: FloatLike = ScheduledFloat((0.0, 0.5), (4000.0, 0.05), default=0),
attention_skip_rate: FloatLike = ScheduledFloat((0.0, 0.2), (4000.0, 0.05), (16000, 0.0), default=0),
conv_skip_rate: FloatLike = ScheduledFloat((0.0, 0.2), (4000.0, 0.05), (16000, 0.0), default=0),
const_attention_rate: FloatLike = ScheduledFloat((0.0, 0.1), (4000.0, 0.01), default=0),
const_attention_rate: FloatLike = ScheduledFloat((0.0, 0.25), (4000.0, 0.025), default=0),
bypass_min: FloatLike = ScheduledFloat((0.0, 0.75), (20000.0, 0.2), default=0),
bypass_max: FloatLike = 1.0,
) -> None: