From 5294e51c984f340d60cf8ec5133d49ff6b84b213 Mon Sep 17 00:00:00 2001
From: sandyhouse <lilong12@baidu.com>
Date: Wed, 3 Mar 2021 10:50:43 +0800
Subject: [PATCH] update

---
 paddle/fluid/framework/trainer_desc.proto | 1 +
 paddle/fluid/pybind/pybind.cc             | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/framework/trainer_desc.proto b/paddle/fluid/framework/trainer_desc.proto
index 1bcdc8458c7..504885ff5cc 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 d812203c041..b3d9e22dba8 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);
                }
              }
 
-- 
GitLab