mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Bug fix in ScheduledSampler

Browse Source
This commit is contained in:
Daniel Povey 2022-11-14 13:52:14 +08:00
parent 614b5b1a52
commit 4c8575878a
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py
View File

@ -1023,6 +1023,7 @@ class ScheduledFloat(torch.nn.Module):
next_x, next_y = self.schedule[i] next_x, next_y = self.schedule[i]
if batch_count >= cur_x and batch_count <= next_x: if batch_count >= cur_x and batch_count <= next_x:
return cur_y + (next_y - cur_y) * (batch_count - cur_x) / (next_x - cur_x) return cur_y + (next_y - cur_y) * (batch_count - cur_x) / (next_x - cur_x)
cur_x, cur_y = next_x, next_y
assert False assert False
FloatLike = Union[float, ScheduledFloat] FloatLike = Union[float, ScheduledFloat]