mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Remove debug print
This commit is contained in:
parent
d137118484
commit
679ba2ee5e
@ -244,7 +244,6 @@ class SoftmaxFunction(torch.autograd.Function):
|
|||||||
|
|
||||||
def softmax(x: Tensor,
|
def softmax(x: Tensor,
|
||||||
dim: int):
|
dim: int):
|
||||||
logging.info(f"torch.is_autocast_enabled()={torch.is_autocast_enabled()}, x dtype={x.dtype}")
|
|
||||||
return SoftmaxFunction.apply(x, dim)
|
return SoftmaxFunction.apply(x, dim)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user