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
|
@staticmethod
|
||||||
@custom_bwd
|
@custom_bwd
|
||||||
def backward(ctx, y_grad: Tensor) -> Tuple[Tensor, None]:
|
def backward(ctx, y_grad: Tensor):
|
||||||
x, y = ctx.saved_tensors
|
x, y = ctx.saved_tensors
|
||||||
x.requires_grad = ctx.x_requires_grad
|
x.requires_grad = ctx.x_requires_grad
|
||||||
m = ctx.m # e.g. a nn.Module
|
m = ctx.m # e.g. a nn.Module
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user