p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!3321 fix wide and deep performance 

Merge pull request !3321 from anzhengqi/fix-wide-deep-performance
This commit is contained in:
mindspore-ci-bot 2020-07-22 17:14:55 +08:00 committed by Gitee
parent 8f4bab4e75 f3ae371436
commit 362a14e008
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model_zoo/official/recommend/wide_and_deep/train_and_eval_distribute.py
View File

@ -110,7 +110,8 @@ def train_and_eval(config):
out = model.eval(ds_eval)
print("=====" * 5 + "model.eval() initialized: {}".format(out))
model.train(epochs, ds_train,
callbacks=[TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback, ckpoint_cb])
callbacks=[TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback, ckpoint_cb],
sink_size=ds_train.get_dataset_size())
if __name__ == "__main__":