diff --git a/paddle/fluid/framework/data_feed.cc b/paddle/fluid/framework/data_feed.cc index 6b0404134d0c07e59102361156d9f1023d817156..7529b7578e607a49395d9f486ac7a5e20cf7aa69 100644 --- a/paddle/fluid/framework/data_feed.cc +++ b/paddle/fluid/framework/data_feed.cc @@ -546,7 +546,7 @@ bool MultiSlotDataFeed::CheckFile(const char* filename) { VLOG(0) << "please check line<" << instance_cout << "> in file<" << filename << ">"; return false; - } else if (errno == ERANGE) { + } else if (errno == ERANGE || num > INT_MAX) { VLOG(0) << "error: the number of ids greater than INT_MAX"; VLOG(0) << "please check line<" << instance_cout << "> in file<" << filename << ">";