diff --git a/PyTorch/built-in/nlp/Bert-Squad_ID0470_for_PyTorch/run_squad.py b/PyTorch/built-in/nlp/Bert-Squad_ID0470_for_PyTorch/run_squad.py index 4c8448f9e08db521d9e732f07c17f43d8600545c..c46721df1c6d68d9543aefced7e6967d9a2cef63 100644 --- a/PyTorch/built-in/nlp/Bert-Squad_ID0470_for_PyTorch/run_squad.py +++ b/PyTorch/built-in/nlp/Bert-Squad_ID0470_for_PyTorch/run_squad.py @@ -31,7 +31,7 @@ import numpy as np import torch if torch.__version__ >= "1.8": import torch_npu - torch.npu.config.allow_internal_format = True + torch.npu.config.allow_internal_format = False from torch.utils.data import (DataLoader, RandomSampler, SequentialSampler, TensorDataset) from torch.utils.data.distributed import DistributedSampler