From 8d9301e225f75cbc94e181dfc81fa0317ea7be76 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Thu, 15 Dec 2022 23:47:41 +0800 Subject: [PATCH] Remove potentially wrong typing info --- 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 51a0f99e8..13d2d890e 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py @@ -252,7 +252,7 @@ class CachingEvalFunction(torch.autograd.Function): @staticmethod @custom_bwd - def backward(ctx, y_grad: Tensor) -> Tuple[Tensor, None]: + def backward(ctx, y_grad: Tensor): x, y = ctx.saved_tensors x.requires_grad = ctx.x_requires_grad m = ctx.m # e.g. a nn.Module