1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- syntax = "proto3";
- import "easyfl/pb/common.proto";
- package easyfl.pb;
- enum OperationType {
- OP_TYPE_TRAIN = 0;
- OP_TYPE_TEST = 1;
- }
- message OperateRequest {
-
- OperationType type = 1;
-
- bytes model = 2;
-
-
- int32 data_index = 3;
-
- OperateConfig config = 4;
- }
- message OperateConfig {
- int32 batch_size = 1;
- int32 local_epoch = 2;
- int64 seed = 3;
- Optimizer optimizer = 4;
-
- bool local_test = 5;
-
- string task_id = 6;
-
- int32 round_id = 7;
-
- bool track = 8;
-
- int32 test_batch_size = 9;
- }
- message Optimizer {
- string type = 1;
- float lr = 2;
- float momentum = 3;
- }
- message OperateResponse {
- Status status = 1;
- }
- service ClientService {
- rpc Operate (OperateRequest) returns (OperateResponse) {}
- }
|