diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py index ced3b96d1..f396b5c65 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py @@ -914,7 +914,7 @@ class WithLoss(torch.autograd.Function): ctx.y_shape = y.shape if random.random() < 0.002 and name is not None: loss_sum = y.sum().item() - logging.info(f"WithLoss: name={ctx.name}, loss-sum={loss_sum:.3e}") + logging.info(f"WithLoss: name={name}, loss-sum={loss_sum:.3e}") return x @staticmethod def backward(ctx, ans_grad: Tensor):