diff --git a/paddle/fluid/framework/trainer_desc.proto b/paddle/fluid/framework/trainer_desc.proto
index 6dbce6a02ca1c037e4d5f54e7a6d2462cc68578b..389c1a870fb54ad28806ad49632323b1c93676f4 100644
--- a/paddle/fluid/framework/trainer_desc.proto
+++ b/paddle/fluid/framework/trainer_desc.proto
@@ -39,7 +39,7 @@ message TrainerDesc {
   optional DataFeedDesc data_desc = 201;
 }
 
-message HogwildWorkerParameter {}
+message HogwildWorkerParameter { repeated string skip_ops = 1; }
 
 message DownpourWorkerParameter {
   repeated TableParameter sparse_table = 1;
diff --git a/python/paddle/fluid/trainer_desc.py b/python/paddle/fluid/trainer_desc.py
index 70b511dd7977bb98a996ef8aec73cbedb7f4df65..5a19907ab699e7d9f58f8f589334fddec3c1abaf 100644
--- a/python/paddle/fluid/trainer_desc.py
+++ b/python/paddle/fluid/trainer_desc.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from proto import trainer_desc_pb2
+from proto import trainer_desc_pb2 as trainer_desc_pb2
 from distributed import ps_pb2 as ps_pb2
 from device_worker import DeviceWorkerFactory
 from google.protobuf import text_format