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

fix_copy_if_different
Chen Weihang 5 years ago committed by GitHub
parent fb70682f00
commit f07b25d8e5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -188,6 +188,14 @@ class MultiDeviceFeedReader {
result.emplace_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) {
ret.emplace(names_[j], std::move(ret_[i][j]));
}

Loading…
Cancel
Save