diff --git a/official/nlp/pangu_alpha/train.py b/official/nlp/pangu_alpha/train.py index b7e2d1f7f97e931766fa1ed9d2c01a949c93bfaf..0b72d10fda10ed9b0dbee6fbc6b246329280228e 100644 --- a/official/nlp/pangu_alpha/train.py +++ b/official/nlp/pangu_alpha/train.py @@ -115,7 +115,7 @@ def run_train(args_opt): load_ckpt_path=args_opt.load_ckpt_path, param_init_type=mstype.float32 if args_opt.param_init_type == 'fp32' else mstype.float16, enable_offload=bool(args_opt.opt_offload), - hidden_act='fast_gelu' if args_opt.device_taget != "GPU" else 'gelu', + hidden_act='fast_gelu' if args_opt.device_target != "GPU" else 'gelu', parallel_config=parallel_config) print("===config is: ", config, flush=True) # Define network