From f66c1600f45c190667107c05f9902ac4612a4208 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Thu, 15 Dec 2022 21:55:23 +0800 Subject: [PATCH] Bug fix to printing code --- 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 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):