From f4ce5768f1163be2fee3d43deaa86304faa45cb5 Mon Sep 17 00:00:00 2001 From: huangxinjing Date: Wed, 25 Nov 2020 09:32:15 +0800 Subject: [PATCH] fix eval error in single device and data parallel mode --- model_zoo/official/recommend/wide_and_deep/train_and_eval.py | 2 +- .../recommend/wide_and_deep/train_and_eval_distribute.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/model_zoo/official/recommend/wide_and_deep/train_and_eval.py b/model_zoo/official/recommend/wide_and_deep/train_and_eval.py index ef79f770ef..b88c72f6a8 100644 --- a/model_zoo/official/recommend/wide_and_deep/train_and_eval.py +++ b/model_zoo/official/recommend/wide_and_deep/train_and_eval.py @@ -95,7 +95,7 @@ def test_train_eval(config): ckptconfig = CheckpointConfig(save_checkpoint_steps=ds_train.get_dataset_size(), keep_checkpoint_max=5) ckpoint_cb = ModelCheckpoint(prefix='widedeep_train', directory=config.ckpt_path, config=ckptconfig) - out = model.eval(ds_eval) + out = model.eval(ds_eval, dataset_sink_mode=(not sparse)) print("=====" * 5 + "model.eval() initialized: {}".format(out)) model.train(epochs, ds_train, callbacks=[TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback, ckpoint_cb], 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 5bf5720b66..1f8ad32f06 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 @@ -105,7 +105,7 @@ def train_and_eval(config): ckpoint_cb = ModelCheckpoint(prefix='widedeep_train', directory=config.ckpt_path + '/ckpt_' + str(get_rank()) + '/', config=ckptconfig) - out = model.eval(ds_eval) + out = model.eval(ds_eval, dataset_sink_mode=(not sparse)) print("=====" * 5 + "model.eval() initialized: {}".format(out)) callback_list = [TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback] if get_rank() == 0: