Update model.py

This commit is contained in:
Mingshuang Luo 2021-11-10 11:38:50 +08:00
parent d2fe504163
commit 917e4868d6

View File

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