提交 5294e51c 编写于 作者: S sandyhouse

update

上级 8b170ffa
...@@ -95,6 +95,7 @@ message SectionWorkerParameter { ...@@ -95,6 +95,7 @@ message SectionWorkerParameter {
optional int32 num_microbatches = 6; optional int32 num_microbatches = 6;
optional int32 num_pipeline_stages = 7 [ default = 1 ]; optional int32 num_pipeline_stages = 7 [ default = 1 ];
optional int32 pipeline_stage = 8 [ default = 1 ]; optional int32 pipeline_stage = 8 [ default = 1 ];
optional int32 schedule_mode = 9 [ default = 0 ];
} }
message SectionConfig { message SectionConfig {
......
...@@ -1454,7 +1454,7 @@ All parameter, weight, gradient are variables in Paddle. ...@@ -1454,7 +1454,7 @@ All parameter, weight, gradient are variables in Paddle.
"number on your machine is %d", "number on your machine is %d",
dev_id, platform::GetCUDADeviceCount(), dev_id, platform::GetCUDADeviceCount(),
platform::GetCUDADeviceCount()); platform::GetCUDADeviceCount());
// std::exit(-1); std::exit(-1);
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册