diff --git a/paddle/fluid/framework/trainer_desc.proto b/paddle/fluid/framework/trainer_desc.proto index 1bcdc8458c7d0ece664d352201c93c95bf834111..504885ff5ccbce760c0a659aedabef6790de5f1a 100644 --- a/paddle/fluid/framework/trainer_desc.proto +++ b/paddle/fluid/framework/trainer_desc.proto @@ -95,6 +95,7 @@ message SectionWorkerParameter { optional int32 num_microbatches = 6; optional int32 num_pipeline_stages = 7 [ default = 1 ]; optional int32 pipeline_stage = 8 [ default = 1 ]; + optional int32 schedule_mode = 9 [ default = 0 ]; } message SectionConfig { diff --git a/paddle/fluid/pybind/pybind.cc b/paddle/fluid/pybind/pybind.cc index d812203c041ac1f2caa5b4685bdf48b8766fe1a0..b3d9e22dba8d244f5e57267a527de0bbdc534996 100644 --- a/paddle/fluid/pybind/pybind.cc +++ b/paddle/fluid/pybind/pybind.cc @@ -1454,7 +1454,7 @@ All parameter, weight, gradient are variables in Paddle. "number on your machine is %d", dev_id, platform::GetCUDADeviceCount(), platform::GetCUDADeviceCount()); - // std::exit(-1); + std::exit(-1); } }