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

!14446 [MD] TFRecord parse failed without print file name 

From: @xiefangqi
Reviewed-by: @liucunwei,@heleiwang
Signed-off-by: @liucunwei
This commit is contained in:
mindspore-ci-bot 2021-03-31 14:40:46 +08:00 committed by Gitee
parent 3ac935344e a81f0fbbfa
commit b27333ba35
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/minddata/dataset/engine/datasetops/source/tf_reader_op.cc
View File

@ -589,7 +589,8 @@ Status TFReaderOp::LoadFile(const std::string &filename, const int64_t start_off
if (start_offset == kInvalidOffset || (rows_total >= start_offset && rows_total < end_offset)) { if (start_offset == kInvalidOffset || (rows_total >= start_offset && rows_total < end_offset)) {
dataengine::Example tf_file; dataengine::Example tf_file;
if (!tf_file.ParseFromString(serialized_example)) { if (!tf_file.ParseFromString(serialized_example)) {
std::string errMsg = "Invalid file, failed to parse tfrecord file : " + serialized_example; std::string errMsg = "Invalid file, failed to parse tfrecord file : " + filename;
MS_LOG(DEBUG) << errMsg + ", details of string: " << serialized_example;
RETURN_STATUS_UNEXPECTED(errMsg); RETURN_STATUS_UNEXPECTED(errMsg);
} }
int32_t num_columns = data_schema_->NumColumns(); int32_t num_columns = data_schema_->NumColumns();