From 2f1651b97e7ac168d1e5856f45b3f94c7c3609ca Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=83=91=E7=89=B9=E9=A9=B9?= Date: Wed, 18 Jun 2025 11:35:56 +0800 Subject: [PATCH] =?UTF-8?q?[built-in][Pytorch][ESPNet2]=20PyTorch=5F2.6?= =?UTF-8?q?=E4=BB=A5=E5=8F=8APython=5F3.11=E9=80=82=E9=85=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../audio/ESPnet2_for_PyTorch/espnet2/train/trainer.py | 4 ++++ PyTorch/built-in/audio/ESPnet2_for_PyTorch/setup.py | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) 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 29257c02d0..ad23879faa 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 87c6ca8aef..160a7b3e95 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", ], -- Gitee