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

fix for memory usage in pruned_transducer_stateless7/scaling.py (#752)

Browse Source 
Co-authored-by: yifanyang <yifanyeung@yifanyangs-MacBook-Pro.local>
This commit is contained in:
Yifan Yang 2022-12-09 19:23:11 +08:00 committed by GitHub
parent 4501821fd9
commit a0cf85343d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py
View File

@ -562,7 +562,7 @@ class ActivationBalancer(torch.nn.Module):
sign_factor = None sign_factor = None
scale_factor = _compute_scale_factor( scale_factor = _compute_scale_factor(
x, x.detach(),
self.channel_dim, self.channel_dim,
min_abs=self.min_abs, min_abs=self.min_abs,
max_abs=self.max_abs, max_abs=self.max_abs,