From a0cf85343dad31a678ddaac7652f0bb2bbb4cac2 Mon Sep 17 00:00:00 2001 From: Yifan Yang <64255737+yfyeung@users.noreply.github.com> Date: Fri, 9 Dec 2022 19:23:11 +0800 Subject: [PATCH] fix for memory usage in pruned_transducer_stateless7/scaling.py (#752) Co-authored-by: yifanyang --- egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py index 6f63e0629..042c9c3e4 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py @@ -562,7 +562,7 @@ class ActivationBalancer(torch.nn.Module): sign_factor = None scale_factor = _compute_scale_factor( - x, + x.detach(), self.channel_dim, min_abs=self.min_abs, max_abs=self.max_abs,