Browse Source

set different ckpt file name for different rank id

tags/v1.1.0
lizhenyu 5 years ago
parent
commit
b41892d728
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      model_zoo/official/recommend/wide_and_deep/train_and_eval_parameter_server_distribute.py

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

@@ -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:


Loading…
Cancel
Save