Reduce final_layerdrop_rate coefficient.

This commit is contained in:
Daniel Povey 2023-01-12 20:00:49 +08:00
parent 3fdfec1049
commit 65f15c9d14

View File

@ -198,7 +198,7 @@ class Zipformer(EncoderInterface):
dropout=dropout, dropout=dropout,
warmup_begin=warmup_batches * (i + 1) / (num_encoders + 1), warmup_begin=warmup_batches * (i + 1) / (num_encoders + 1),
warmup_end=warmup_batches * (i + 2) / (num_encoders + 1), warmup_end=warmup_batches * (i + 2) / (num_encoders + 1),
final_layerdrop_rate=0.05 * (downsampling_factor[i] ** 0.5), final_layerdrop_rate=0.035 * (downsampling_factor[i] ** 0.5),
attention_share_layers=attention_share_layers[i], attention_share_layers=attention_share_layers[i],
) )