mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 18:24:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update model.py

Browse Source
This commit is contained in:
Mingshuang Luo 2021-11-10 11:38:50 +08:00
parent d2fe504163
commit 917e4868d6
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/timit/ASR/tdnn_ligru_ctc/model.py
View File

@ -452,8 +452,9 @@ class LiGRU_Layer(torch.nn.Module):
).data ).data
# Sampling the mask # Sampling the mask
left_boundary = self.drop_mask_cnt
right_boundary = self.drop_mask_cnt + self.batch_size right_boundary = self.drop_mask_cnt + self.batch_size
drop_mask = self.drop_masks[self.drop_mask_cnt:right_boundary] drop_mask = self.drop_masks[left_boundary:right_boundary]
self.drop_mask_cnt = self.drop_mask_cnt + self.batch_size self.drop_mask_cnt = self.drop_mask_cnt + self.batch_size
else: else: