update load_error_info, test=develop (#18000)

Repair error prompt: Users are prompted to check whether the model or parameter files are damaged when loading parameters are wrong.
test
lujun 6 years ago committed by GitHub
parent 04ea7cb069
commit 75fcd29220
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -65,7 +65,10 @@ class LoadCombineOpKernel : public framework::OpKernel<T> {
auto *tensor = out_vars[i]->GetMutable<framework::LoDTensor>();
// Error checking
PADDLE_ENFORCE(static_cast<bool>(*buffer), "Cannot read more");
PADDLE_ENFORCE(
static_cast<bool>(*buffer),
"There is a problem with loading model parameters. "
"Please check whether the model file is complete or damaged.");
// Get data from fin to tensor
DeserializeFromStream(*buffer, tensor, dev_ctx);

Loading…
Cancel
Save