diff --git a/PyTorch/built-in/audio/ESPnet2_for_PyTorch/espnet2/train/trainer.py b/PyTorch/built-in/audio/ESPnet2_for_PyTorch/espnet2/train/trainer.py index 29257c02d0761ebecc2a6dc8c8484c72db679183..ad23879faa8eda4d84ef9879908e2808c37367f0 100644 --- a/PyTorch/built-in/audio/ESPnet2_for_PyTorch/espnet2/train/trainer.py +++ b/PyTorch/built-in/audio/ESPnet2_for_PyTorch/espnet2/train/trainer.py @@ -36,6 +36,10 @@ import numpy as np import torch if torch.__version__ >= '1.8': import torch_npu + +if torch.__version__ >= '2.6': + torch._C._set_math_sdp_allow_fp16_bf16_reduction(True) + import torch.nn import torch.optim from typeguard import check_argument_types diff --git a/PyTorch/built-in/audio/ESPnet2_for_PyTorch/setup.py b/PyTorch/built-in/audio/ESPnet2_for_PyTorch/setup.py index 87c6ca8aef2b405773c0979881e3d5b1649569cc..160a7b3e954065f88427ddadb15ae15c048a6f37 100644 --- a/PyTorch/built-in/audio/ESPnet2_for_PyTorch/setup.py +++ b/PyTorch/built-in/audio/ESPnet2_for_PyTorch/setup.py @@ -40,7 +40,7 @@ requirements = { "train": [ "matplotlib==3.1.0", "pillow>=6.1.0", - "editdistance==0.5.2", + "editdistance==0.6.2", "wandb", "tensorboard>=1.14", ],