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

!49235 Refine error log for embedding cache mode 

Merge pull request !49235 from zyli2020/embedding_cache_master
This commit is contained in:
i-robot 2023-02-22 11:27:07 +00:00 committed by Gitee
parent 4e59e010cb fd8745626d
commit 64a07d9666
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/train/model.py
View File

@ -1027,7 +1027,7 @@ class Model:
# Parameter server and embedding cache mode check.
if _is_ps_mode():
if not dataset_sink_mode and _cache_enable():
raise ValueError("Embedding cache mode should run with 'data_sink_mode=True'.")
raise ValueError("Embedding cache mode should run with 'dataset_sink_mode=True'.")
Validator.check_is_int(sink_size)