Remove debug print

This commit is contained in:
Daniel Povey 2022-10-20 13:30:55 +08:00
parent d137118484
commit 679ba2ee5e

View File

@ -244,7 +244,6 @@ class SoftmaxFunction(torch.autograd.Function):
def softmax(x: Tensor,
dim: int):
logging.info(f"torch.is_autocast_enabled()={torch.is_autocast_enabled()}, x dtype={x.dtype}")
return SoftmaxFunction.apply(x, dim)