Update model.py

This commit is contained in:
Mingshuang Luo 2021-11-09 22:40:39 +08:00
parent 1193228a14
commit 55f9bbdb15

View File

@ -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