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