From b41892d72860873587d4c670ecf46fd9441da370 Mon Sep 17 00:00:00 2001 From: lizhenyu Date: Mon, 21 Dec 2020 15:19:20 +0800 Subject: [PATCH] set different ckpt file name for different rank id --- .../wide_and_deep/train_and_eval_parameter_server_distribute.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model_zoo/official/recommend/wide_and_deep/train_and_eval_parameter_server_distribute.py b/model_zoo/official/recommend/wide_and_deep/train_and_eval_parameter_server_distribute.py index b87352088e..21b211bb04 100644 --- a/model_zoo/official/recommend/wide_and_deep/train_and_eval_parameter_server_distribute.py +++ b/model_zoo/official/recommend/wide_and_deep/train_and_eval_parameter_server_distribute.py @@ -126,6 +126,7 @@ def train_and_eval(config): directory=config.ckpt_path + '/ckpt_' + str(get_rank()) + '/', config=ckptconfig) if cache_enable: + config.stra_ckpt = './stra_ckpt_' + str(get_rank()) + '/strategy.ckpt' context.set_auto_parallel_context(strategy_ckpt_save_file=config.stra_ckpt) callback_list = [TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback] if get_rank() == 0: