mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix issue with warmup in test time

Browse Source
This commit is contained in:
Daniel Povey 2022-10-08 11:01:02 +08:00
parent 97bc894f62
commit 9c1a239931
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py
View File

@ -401,7 +401,9 @@ class ConformerEncoder(nn.Module):
batch = self.count.item()
if self.training:
self.count += 1
return min(1.0, batch / self.warmup_batches)
return min(1.0, batch / self.warmup_batches)
else:
return 1.0 # this is mostly a workaround for an issue with moderl averaging.
def forward(