m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Davinci to Ascend

Browse Source
pull/6135/head
yao_yf 4 years ago
parent 0251de8482
commit 3f00d264fd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
model_zoo/official/recommend/wide_and_deep_multitable/train_and_eval_distribute.py
Unescape View File

@ -110,7 +110,7 @@ if __name__ == "__main__":
wide_and_deep_config = WideDeepConfig() wide_and_deep_config = WideDeepConfig()
wide_and_deep_config.argparse_init() wide_and_deep_config.argparse_init()
compute_emb_dim(wide_and_deep_config) 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) save_graphs=True)
init() init()
context.set_auto_parallel_context(parallel_mode=ParallelMode.DATA_PARALLEL, gradients_mean=True, context.set_auto_parallel_context(parallel_mode=ParallelMode.DATA_PARALLEL, gradients_mean=True,

Write Preview
Loading…
Cancel
Save