mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-08 08:34:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix

Browse Source
This commit is contained in:
Daniel Povey 2022-05-18 10:26:29 +08:00
parent 668d01cc7a
commit ebc2ffeff7
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless4/optim.py
View File

@ -632,7 +632,7 @@ class Cain(Optimizer):
rev_dims_order.append(ndim-1) rev_dims_order.append(ndim-1)
for i in range(dim+1, ndim): for i in range(dim+1, ndim):
dims_order.append(i) dims_order.append(i)
rev_dims_order.append(i) rev_dims_order.append(i-1)
dims_order.append(dim) dims_order.append(dim)
# e.g. ndim=4, dim=1, dims_order=(0,2,3,1), rev_dims_order=(0,3,1,2) # e.g. ndim=4, dim=1, dims_order=(0,2,3,1), rev_dims_order=(0,3,1,2)
new_grad = grad.permute(*dims_order) new_grad = grad.permute(*dims_order)

2
egs/librispeech/ASR/pruned_transducer_stateless4b/optim2.py
View File

@ -628,7 +628,7 @@ class Eve(Optimizer):
rev_dims_order.append(ndim-1) rev_dims_order.append(ndim-1)
for i in range(dim+1, ndim): for i in range(dim+1, ndim):
dims_order.append(i) dims_order.append(i)
rev_dims_order.append(i) rev_dims_order.append(i-1)
dims_order.append(dim) dims_order.append(dim)
# e.g. ndim=4, dim=1, dims_order=(0,2,3,1), rev_dims_order=(0,3,1,2) # e.g. ndim=4, dim=1, dims_order=(0,2,3,1), rev_dims_order=(0,3,1,2)
new_grad = grad.permute(*dims_order) new_grad = grad.permute(*dims_order)