Browse Source

add detail batch error to log

tags/v0.3.0-alpha
xiefangqi 5 years ago
parent
commit
25bb3960d4
1 changed files with 8 additions and 1 deletions
  1. +8
    -1
      mindspore/ccsrc/dataset/engine/datasetops/batch_op.cc

+ 8
- 1
mindspore/ccsrc/dataset/engine/datasetops/batch_op.cc View File

@@ -164,7 +164,14 @@ Status BatchOp::BatchRows(const std::unique_ptr<TensorQTable> *source_table,
if (row[i]->shape() == row_shapes[i]) { // check the newly popped rows have the same dim as the first
RETURN_IF_NOT_OK(batched_row[i]->InsertTensor(std::vector<dsize_t>(1, j), row[i]));
} else {
RETURN_STATUS_UNEXPECTED("[Batch ERROR] Inconsistent TensorShapes\n");
std::string column_name;
for (auto itr : column_name_id_map_) {
if (static_cast<size_t>(itr.second) == i) {
column_name = itr.first;
break;
}
}
RETURN_STATUS_UNEXPECTED("[Batch ERROR] Inconsistent TensorShapes of Column " + column_name);
}
}
}


Loading…
Cancel
Save