mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Merge branch 'scaled_adam_exp149' into scaled_adam_exp150
This commit is contained in:
commit
5a0914fdcf
@ -282,7 +282,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)
|
||||
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user