|
|
@ -47,7 +47,9 @@ void Init(const std::vector<std::string> argv) {
|
|
|
|
|
|
|
|
|
|
|
|
void ReadBinaryFile(const std::string& filename, std::string* contents) {
|
|
|
|
void ReadBinaryFile(const std::string& filename, std::string* contents) {
|
|
|
|
std::ifstream fin(filename, std::ios::in | std::ios::binary);
|
|
|
|
std::ifstream fin(filename, std::ios::in | std::ios::binary);
|
|
|
|
PADDLE_ENFORCE(static_cast<bool>(fin), "Cannot open file %s", filename);
|
|
|
|
PADDLE_ENFORCE_EQ(
|
|
|
|
|
|
|
|
fin.is_open(), true,
|
|
|
|
|
|
|
|
platform::errors::Unavailable("Failed to open file %s.", filename));
|
|
|
|
fin.seekg(0, std::ios::end);
|
|
|
|
fin.seekg(0, std::ios::end);
|
|
|
|
contents->clear();
|
|
|
|
contents->clear();
|
|
|
|
contents->resize(fin.tellg());
|
|
|
|
contents->resize(fin.tellg());
|
|
|
@ -133,9 +135,10 @@ std::unique_ptr<framework::ProgramDesc> Load(framework::Executor* executor,
|
|
|
|
|
|
|
|
|
|
|
|
std::unique_ptr<framework::ProgramDesc> main_program(
|
|
|
|
std::unique_ptr<framework::ProgramDesc> main_program(
|
|
|
|
new framework::ProgramDesc(program_desc_str));
|
|
|
|
new framework::ProgramDesc(program_desc_str));
|
|
|
|
PADDLE_ENFORCE(framework::IsProgramVersionSupported(main_program->Version()),
|
|
|
|
PADDLE_ENFORCE_EQ(
|
|
|
|
"model version %ld is not supported.",
|
|
|
|
framework::IsProgramVersionSupported(main_program->Version()), true,
|
|
|
|
main_program->Version());
|
|
|
|
platform::errors::Unavailable("Model version %ld is not supported.",
|
|
|
|
|
|
|
|
main_program->Version()));
|
|
|
|
|
|
|
|
|
|
|
|
// model_from_memory is false in separate parameters.
|
|
|
|
// model_from_memory is false in separate parameters.
|
|
|
|
LoadPersistables(executor, scope, *main_program, dirname, "",
|
|
|
|
LoadPersistables(executor, scope, *main_program, dirname, "",
|
|
|
@ -151,9 +154,10 @@ std::unique_ptr<framework::ProgramDesc> Load(
|
|
|
|
|
|
|
|
|
|
|
|
std::unique_ptr<framework::ProgramDesc> main_program(
|
|
|
|
std::unique_ptr<framework::ProgramDesc> main_program(
|
|
|
|
new framework::ProgramDesc(program_desc_str));
|
|
|
|
new framework::ProgramDesc(program_desc_str));
|
|
|
|
PADDLE_ENFORCE(framework::IsProgramVersionSupported(main_program->Version()),
|
|
|
|
PADDLE_ENFORCE_EQ(
|
|
|
|
"model version %ld is not supported.",
|
|
|
|
framework::IsProgramVersionSupported(main_program->Version()), true,
|
|
|
|
main_program->Version());
|
|
|
|
platform::errors::Unavailable("Model version %ld is not supported.",
|
|
|
|
|
|
|
|
main_program->Version()));
|
|
|
|
|
|
|
|
|
|
|
|
LoadPersistables(executor, scope, *main_program, "", param_filename,
|
|
|
|
LoadPersistables(executor, scope, *main_program, "", param_filename,
|
|
|
|
false /* model_from_memory */);
|
|
|
|
false /* model_from_memory */);
|
|
|
@ -165,9 +169,10 @@ std::unique_ptr<framework::ProgramDesc> LoadFromMemory(
|
|
|
|
const std::string& prog_buffer, const std::string& param_buffer) {
|
|
|
|
const std::string& prog_buffer, const std::string& param_buffer) {
|
|
|
|
std::unique_ptr<framework::ProgramDesc> main_program(
|
|
|
|
std::unique_ptr<framework::ProgramDesc> main_program(
|
|
|
|
new framework::ProgramDesc(prog_buffer));
|
|
|
|
new framework::ProgramDesc(prog_buffer));
|
|
|
|
PADDLE_ENFORCE(framework::IsProgramVersionSupported(main_program->Version()),
|
|
|
|
PADDLE_ENFORCE_EQ(
|
|
|
|
"model version %ld is not supported.",
|
|
|
|
framework::IsProgramVersionSupported(main_program->Version()), true,
|
|
|
|
main_program->Version());
|
|
|
|
platform::errors::Unavailable("Model version %ld is not supported.",
|
|
|
|
|
|
|
|
main_program->Version()));
|
|
|
|
|
|
|
|
|
|
|
|
LoadPersistables(executor, scope, *main_program, "", param_buffer,
|
|
|
|
LoadPersistables(executor, scope, *main_program, "", param_buffer,
|
|
|
|
true /* model_filename */);
|
|
|
|
true /* model_filename */);
|
|
|
|