未验证 提交 f07b25d8 编写于 作者: C Chen Weihang 提交者: GitHub

fix DataLoader.generrator using error, test=develop (#25355)

上级 fb70682f
...@@ -188,6 +188,14 @@ class MultiDeviceFeedReader { ...@@ -188,6 +188,14 @@ class MultiDeviceFeedReader {
result.emplace_back(); result.emplace_back();
auto &ret = result.back(); auto &ret = result.back();
PADDLE_ENFORCE_EQ(names_.size(), ret_[i].size(),
platform::errors::InvalidArgument(
"The sample number of reader's input data and the "
"input number of feed list are not equal.\n"
"Possible reasons are:\n"
" The generator is decorated by `paddle.batch` "
"and configured by `set_batch_generator`, but here "
"need to used `set_sample_list_generator`."));
for (size_t j = 0; j < names_.size(); ++j) { for (size_t j = 0; j < names_.size(); ++j) {
ret.emplace(names_[j], std::move(ret_[i][j])); ret.emplace(names_[j], std::move(ret_[i][j]));
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册