From 251bff3f41d84b0e635618fcbfc7dbe1859175e4 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sat, 18 Jun 2022 08:54:03 +0800 Subject: [PATCH] Fix an error introduced by supporting torchscript for torch 1.6.0 --- egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py index 8bd50d185..c190be626 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py @@ -56,10 +56,10 @@ class ActivationBalancerFunction(torch.autograd.Function): # sum_dims = [d for d in range(x.ndim) if d != channel_dim] # The above line is not torch scriptable for torch 1.6.0 # torch.jit.frontend.NotSupportedError: comprehension ifs not supported yet: # noqa - sum_dims = 0 + sum_dims = [] for d in range(x.ndim): if d != channel_dim: - sum_dims += d + sum_dims.append(d) xgt0 = x > 0 proportion_positive = torch.mean(