-
Notifications
You must be signed in to change notification settings - Fork 4
/
message.proto
76 lines (58 loc) · 1.22 KB
/
message.proto
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
syntax = "proto3";
package message;
// The information service definition.
service Information{
// Sends an ID
rpc InsertTrajectory (Trajectory) returns (InsertTrajectoryReply) {}
rpc GetParams (GetParamsRequest) returns (ModelParams) {}
rpc Quit (QuitRequest) returns (QuitReply) {}
}
message InsertTrajectoryReply {
string message = 1;
}
message GetParamsRequest {
}
message QuitRequest {
}
message QuitReply {
}
message GetParamsReply {
int32 frame_count = 1;
string params = 2;
}
message AgentOut {
TensorInt32 action = 1;
TensorFloat policy_logits = 2;
TensorFloat values = 3;
}
message Timestep {
TensorFloat discount = 1;
TensorFloat observation = 2;
TensorFloat reward = 3;
TensorInt32 step_type = 4;
}
message TensorFloat {
repeated float data = 1;
repeated int32 shape = 2;
}
message TensorInt32 {
repeated float data = 1;
repeated int32 shape = 2;
}
message NamedTuple {
repeated float TensorFloat = 1;
}
message Trajectory {
AgentOut agent_out = 1;
TensorFloat agent_state = 2;
Timestep timestep = 3;
}
message LayerWeight {
string name = 1;
TensorFloat b = 2;
TensorFloat w = 3;
}
message ModelParams {
int32 frame_count = 1;
repeated LayerWeight params = 2;
}