Browse Source

!11758 support recording dict value as string for dataset_graph

From: @luopengting
Reviewed-by: @ouwenchang,@wenkai_dist
Signed-off-by: @wenkai_dist
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
3795cd37d9
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      mindspore/train/callback/_dataset_graph.py

+ 6
- 0
mindspore/train/callback/_dataset_graph.py View File

@@ -13,6 +13,7 @@
# limitations under the License.
# ============================================================================
"""Define dataset graph related operations."""
import json
from importlib import import_module

from mindspore import log as logger
@@ -124,6 +125,11 @@ class DatasetGraph:
if value:
replace_value_list = list(map(lambda x: "" if x is None else x, value))
message.mapStrList[key].strValue.extend(replace_value_list)
elif isinstance(value, dict):
try:
message.mapStr[key] = json.dumps(value)
except TypeError as exo:
logger.warning("Transform the value of parameter %r to string failed. Detail: %s.", key, str(exo))
elif value is None:
message.mapStr[key] = "None"
else:


Loading…
Cancel
Save