diff --git a/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/run_mlm.py b/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/run_mlm.py index 35ce92029a62ab8ed05908a3ee5e917b9fdc8691..ef73e4204b9ed61e7bd2f98f04e3357f44d24022 100644 --- a/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/run_mlm.py +++ b/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/run_mlm.py @@ -214,6 +214,11 @@ def main(): # or by passing the --help flag to this script. # We now keep distinct sets of args, for a cleaner separation of concerns. + # PyTorch1.11版本可能存在精度下降的问题,对算子采取规避方式临时处理的方案 + option = {} + option['NPU_FUZZY_COMPILE_BLACKLIST'] = 'BroadcastTo' + torch.npu.set_option(option) + parser = HfArgumentParser((ModelArguments, DataTrainingArguments, TrainingArguments)) if len(sys.argv) == 2 and sys.argv[1].endswith(".json"): # If we pass only one argument to the script and it's the path to a json file,