From 0611f6231aee44751626750240dd43200dde289c Mon Sep 17 00:00:00 2001 From: lizhenyu Date: Wed, 14 Oct 2020 11:32:54 +0800 Subject: [PATCH] [bugfix] save ir file failed --- .../recommend/wide_and_deep/train_and_eval_distribute.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model_zoo/official/recommend/wide_and_deep/train_and_eval_distribute.py b/model_zoo/official/recommend/wide_and_deep/train_and_eval_distribute.py index 32dc222c01..2fba1fe999 100644 --- a/model_zoo/official/recommend/wide_and_deep/train_and_eval_distribute.py +++ b/model_zoo/official/recommend/wide_and_deep/train_and_eval_distribute.py @@ -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())