mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Remove potentially wrong typing info
This commit is contained in:
parent
6caaa4e9c6
commit
8d9301e225
@ -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
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user