!6135 targrt Davinci to Ascend

Merge pull request !6135 from yao_yf/wide_and_deep_Davinci_to_Ascend
5 years ago · 411f13bde9
parent 144998894f 3f00d264fd
commit 411f13bde9
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/official/recommend/wide_and_deep_multitable/train_and_eval_distribute.py
+++ b/model_zoo/official/recommend/wide_and_deep_multitable/train_and_eval_distribute.py
@ -110,7 +110,7 @@ if __name__ == "__main__":
    wide_and_deep_config = WideDeepConfig()
    wide_and_deep_config.argparse_init()
    compute_emb_dim(wide_and_deep_config)
-    context.set_context(mode=context.GRAPH_MODE, device_target="Davinci",
+    context.set_context(mode=context.GRAPH_MODE, device_target="Ascend",
                        save_graphs=True)
    init()
    context.set_auto_parallel_context(parallel_mode=ParallelMode.DATA_PARALLEL, gradients_mean=True,