From 55f9bbdb1560d1e3bae2055724d9dea98dee207e Mon Sep 17 00:00:00 2001 From: Mingshuang Luo <37799481+luomingshuang@users.noreply.github.com> Date: Tue, 9 Nov 2021 22:40:39 +0800 Subject: [PATCH] Update model.py --- egs/timit/ASR/tdnn_ligru_ctc/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/timit/ASR/tdnn_ligru_ctc/model.py b/egs/timit/ASR/tdnn_ligru_ctc/model.py index 23fc09621..0a4e6d82b 100644 --- a/egs/timit/ASR/tdnn_ligru_ctc/model.py +++ b/egs/timit/ASR/tdnn_ligru_ctc/model.py @@ -453,7 +453,7 @@ class LiGRU_Layer(torch.nn.Module): # Sampling the mask drop_mask = self.drop_masks[ - self.drop_mask_cnt : self.drop_mask_cnt + self.batch_size + self.drop_mask_cnt:self.drop_mask_cnt + self.batch_size ] self.drop_mask_cnt = self.drop_mask_cnt + self.batch_size