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
|
# Sampling the mask
|
||||||
drop_mask = self.drop_masks[
|
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
|
self.drop_mask_cnt = self.drop_mask_cnt + self.batch_size
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user