From 85d4483cd752d80f8caf0cd657c10b7049536e7a Mon Sep 17 00:00:00 2001 From: huangju1993 Date: Thu, 22 Aug 2024 06:35:30 +0000 Subject: [PATCH] update PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/test/train_performance_8p.sh. Signed-off-by: huangju1993 --- .../test/train_performance_8p.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/test/train_performance_8p.sh b/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/test/train_performance_8p.sh index 3fb05e5954..37a63115b8 100644 --- a/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/test/train_performance_8p.sh +++ b/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/test/train_performance_8p.sh @@ -118,7 +118,7 @@ nohup python3 -m torch.distributed.launch --nproc_per_node 8 run_mlm.py \ --fp16_opt_level O2 \ --loss_scale 8192 \ --use_combine_grad \ - --optim adamw_apex_fused_npu \ + --optim adamw_torch_fused \ --distributed_process_group_timeout 5400 \ --output_dir ./output > ${test_path_dir}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & wait -- Gitee