From 0355de9ab29f4cb44527eb7e4dff319babda87ab Mon Sep 17 00:00:00 2001 From: xiefangqi Date: Mon, 17 Aug 2020 22:41:37 +0800 Subject: [PATCH] fix tfrecord issue --- .../dataset/engine/datasetops/source/tf_reader_op.cc | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/tf_reader_op.cc b/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/tf_reader_op.cc index 98c36224151..3ac74247523 100644 --- a/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/tf_reader_op.cc +++ b/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/tf_reader_op.cc @@ -635,7 +635,11 @@ Status TFReaderOp::LoadExample(const dataengine::Example *tf_file, std::unique_p const ColDescriptor current_col = data_schema_->column(col); const dataengine::Features &example_features = tf_file->features(); const google::protobuf::Map &feature_map = example_features.feature(); - const dataengine::Feature &column_values_list = feature_map.at(current_col.name()); + auto iter_column = feature_map.find(current_col.name()); + if (iter_column == feature_map.end()) { + RETURN_STATUS_UNEXPECTED("key not found: " + current_col.name()); + } + const dataengine::Feature &column_values_list = iter_column->second; RETURN_IF_NOT_OK(LoadFeature(tensor_table, column_values_list, current_col, row, col)); }