diff --git a/caffe2/serialize/inline_container.cc b/caffe2/serialize/inline_container.cc index 9847bc132264..9d3cc332ae96 100644 --- a/caffe2/serialize/inline_container.cc +++ b/caffe2/serialize/inline_container.cc @@ -142,7 +142,13 @@ void PyTorchStreamReader::init() { std::tie(version_ptr, version_size) = getRecord("version"); } std::string version(static_cast(version_ptr.get()), version_size); - version_ = caffe2::stoull(version); + try { + version_ = caffe2::stoull(version); + } catch (const std::invalid_argument &e) { + CAFFE_THROW("Couldn't parse the version ", + version, + " as Long Long."); + } // NOLINTNEXTLINE(clang-diagnostic-sign-compare) if (version_ < kMinSupportedFileFormatVersion) { CAFFE_THROW(