fix-develop-build.sh
Xin Pan 7 years ago
parent 56a977d436
commit c69cf6dde8

@ -21,7 +21,7 @@ package paddle.framework.proto;
// //
// Serailization and Deserialization codes should be modified in a way // Serailization and Deserialization codes should be modified in a way
// that supports old versions following the version and compatibility policy. // that supports old versions following the version and compatibility policy.
message Version { optional int64 version = 1 [ default = -1 ]; } message Version { optional int64 version = 1 [ default = 0 ]; }
enum AttrType { enum AttrType {
INT = 0; INT = 0;

@ -125,7 +125,8 @@ 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(framework::IsProgramVersionSupported(main_program->Version()),
"model version %d is not supported.", main_program->Version());
LoadPersistables(executor, scope, *main_program, dirname, ""); LoadPersistables(executor, scope, *main_program, dirname, "");
return main_program; return main_program;
@ -140,7 +141,8 @@ 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(framework::IsProgramVersionSupported(main_program->Version()),
"model version %d is not supported.", main_program->Version());
LoadPersistables(executor, scope, *main_program, "", param_filename); LoadPersistables(executor, scope, *main_program, "", param_filename);
return main_program; return main_program;

Loading…
Cancel
Save