Skip to content
Snippets Groups Projects
Commit 62577788 authored by huangxinjing's avatar huangxinjing
Browse files

Add gelu or fast_gelu according to the device

parent 445536b4
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment