Browse Source

!7280 [bugfix] save ir file failed

Merge pull request !7280 from zyli2020/bug_fix
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
5c86716130
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      model_zoo/official/recommend/wide_and_deep/train_and_eval_distribute.py

+ 1
- 0
model_zoo/official/recommend/wide_and_deep/train_and_eval_distribute.py View File

@@ -120,6 +120,7 @@ if __name__ == "__main__":

context.set_context(mode=context.GRAPH_MODE, device_target=wide_deep_config.device_target, save_graphs=True)
init()
context.set_context(save_graphs_path='./graphs_of_device_id_'+str(get_rank()))
context.set_auto_parallel_context(parallel_mode=ParallelMode.DATA_PARALLEL, gradients_mean=True,
device_num=get_group_size())



Loading…
Cancel
Save