mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Update model.py
This commit is contained in:
parent
1193228a14
commit
55f9bbdb15
@ -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
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user