From 917e4868d68bb83a4413c44da31b965769e01be7 Mon Sep 17 00:00:00 2001 From: Mingshuang Luo <37799481+luomingshuang@users.noreply.github.com> Date: Wed, 10 Nov 2021 11:38:50 +0800 Subject: [PATCH] Update model.py --- egs/timit/ASR/tdnn_ligru_ctc/model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/egs/timit/ASR/tdnn_ligru_ctc/model.py b/egs/timit/ASR/tdnn_ligru_ctc/model.py index 890f48247..64420225b 100644 --- a/egs/timit/ASR/tdnn_ligru_ctc/model.py +++ b/egs/timit/ASR/tdnn_ligru_ctc/model.py @@ -452,8 +452,9 @@ class LiGRU_Layer(torch.nn.Module): ).data # Sampling the mask + left_boundary = self.drop_mask_cnt 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 else: