Browse Source

!4625 fix tf_read_op issue

Merge pull request !4625 from xiefangqi/fix_tfrecord_issue
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
b681ef5fc6
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      mindspore/ccsrc/minddata/dataset/engine/datasetops/source/tf_reader_op.cc

+ 5
- 1
mindspore/ccsrc/minddata/dataset/engine/datasetops/source/tf_reader_op.cc View File

@@ -635,7 +635,11 @@ Status TFReaderOp::LoadExample(const dataengine::Example *tf_file, std::unique_p
const ColDescriptor current_col = data_schema_->column(col); const ColDescriptor current_col = data_schema_->column(col);
const dataengine::Features &example_features = tf_file->features(); const dataengine::Features &example_features = tf_file->features();
const google::protobuf::Map<std::string, dataengine::Feature> &feature_map = example_features.feature(); const google::protobuf::Map<std::string, dataengine::Feature> &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)); RETURN_IF_NOT_OK(LoadFeature(tensor_table, column_values_list, current_col, row, col));
} }




Loading…
Cancel
Save