From 3f00d264fdcb01260a751739b79b2bf7a7fdee61 Mon Sep 17 00:00:00 2001 From: yao_yf Date: Mon, 14 Sep 2020 09:32:40 +0800 Subject: [PATCH] Davinci to Ascend --- .../wide_and_deep_multitable/train_and_eval_distribute.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git 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 index 0f278d1668..c6262336b0 100644 --- 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,