-
Notifications
You must be signed in to change notification settings - Fork 236
/
Copy pathrnn_problems.h
138 lines (134 loc) · 5.61 KB
/
rnn_problems.h
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
// Hidden units, batch size, timesteps, RNN cell type
std::vector<std::tuple<int, int, int, std::string>> training_set = {
std::make_tuple(1760, 16, 50, "vanilla"),
std::make_tuple(1760, 32, 50, "vanilla"),
std::make_tuple(1760, 64, 50, "vanilla"),
std::make_tuple(1760, 128, 50, "vanilla"),
std::make_tuple(2048, 16, 50, "vanilla"),
std::make_tuple(2048, 32, 50, "vanilla"),
std::make_tuple(2048, 64, 50, "vanilla"),
std::make_tuple(2048, 128, 50, "vanilla"),
std::make_tuple(2560, 16, 50, "vanilla"),
std::make_tuple(2560, 32, 50, "vanilla"),
std::make_tuple(2560, 64, 50, "vanilla"),
std::make_tuple(2560, 128, 50, "vanilla"),
std::make_tuple(512, 16, 25, "lstm"),
std::make_tuple(512, 32, 25, "lstm"),
std::make_tuple(512, 64, 25, "lstm"),
std::make_tuple(512, 128, 25, "lstm"),
std::make_tuple(1024, 16, 25, "lstm"),
std::make_tuple(1024, 32, 25, "lstm"),
std::make_tuple(1024, 64, 25, "lstm"),
std::make_tuple(1024, 128, 25, "lstm"),
std::make_tuple(2048, 16, 25, "lstm"),
std::make_tuple(2048, 32, 25, "lstm"),
std::make_tuple(2048, 64, 25, "lstm"),
std::make_tuple(2048, 128, 25, "lstm"),
std::make_tuple(4096, 16, 25, "lstm"),
std::make_tuple(4096, 32, 25, "lstm"),
std::make_tuple(4096, 64, 25, "lstm"),
std::make_tuple(4096, 128, 25, "lstm"),
std::make_tuple(1536, 8, 50, "lstm"),
std::make_tuple(1536, 16, 50, "lstm"),
std::make_tuple(1536, 32, 50, "lstm"),
std::make_tuple(256, 16, 150, "lstm"),
std::make_tuple(256, 32, 150, "lstm"),
std::make_tuple(256, 64, 150, "lstm"),
std::make_tuple(2816, 32, 1500, "gru"),
std::make_tuple(2816, 32, 750, "gru"),
std::make_tuple(2816, 32, 375, "gru"),
std::make_tuple(2816, 32, 187, "gru"),
std::make_tuple(2048, 32, 1500, "gru"),
std::make_tuple(2048, 32, 750, "gru"),
std::make_tuple(2048, 32, 375, "gru"),
std::make_tuple(2048, 32, 187, "gru"),
std::make_tuple(1536, 32, 1500, "gru"),
std::make_tuple(1536, 32, 750, "gru"),
std::make_tuple(1536, 32, 375, "gru"),
std::make_tuple(1536, 32, 187, "gru"),
std::make_tuple(2560, 32, 1500, "gru"),
std::make_tuple(2560, 32, 750, "gru"),
std::make_tuple(2560, 32, 375, "gru"),
std::make_tuple(2560, 32, 187, "gru"),
std::make_tuple(512, 32, 1, "gru"),
std::make_tuple(1024, 32, 1500, "gru"),
std::make_tuple(1024, 64, 1500, "gru")
};
// Hidden units, batch size, timesteps, RNN cell type
std::vector<std::tuple<int, int, int, std::string>> inference_server_set = {
std::make_tuple(512, 1, 25, "lstm"),
std::make_tuple(512, 2, 25, "lstm"),
std::make_tuple(512, 4, 25, "lstm"),
std::make_tuple(1024, 1, 25, "lstm"),
std::make_tuple(1024, 2, 25, "lstm"),
std::make_tuple(1024, 4, 25, "lstm"),
std::make_tuple(2048, 1, 25, "lstm"),
std::make_tuple(2048, 2, 25, "lstm"),
std::make_tuple(2048, 4, 25, "lstm"),
std::make_tuple(1536, 1, 50, "lstm"),
std::make_tuple(1536, 2, 50, "lstm"),
std::make_tuple(1536, 4, 50, "lstm"),
std::make_tuple(256, 1, 150, "lstm"),
std::make_tuple(256, 2, 150, "lstm"),
std::make_tuple(256, 4, 150, "lstm"),
std::make_tuple(2816, 1, 1500, "gru"),
std::make_tuple(2816, 1, 750, "gru"),
std::make_tuple(2816, 1, 375, "gru"),
std::make_tuple(2816, 1, 187, "gru"),
std::make_tuple(2816, 2, 1500, "gru"),
std::make_tuple(2816, 2, 750, "gru"),
std::make_tuple(2816, 2, 375, "gru"),
std::make_tuple(2816, 2, 187, "gru"),
std::make_tuple(2816, 4, 1500, "gru"),
std::make_tuple(2816, 4, 750, "gru"),
std::make_tuple(2816, 4, 375, "gru"),
std::make_tuple(2816, 4, 187, "gru"),
std::make_tuple(2048, 1, 1500, "gru"),
std::make_tuple(2048, 1, 750, "gru"),
std::make_tuple(2048, 1, 375, "gru"),
std::make_tuple(2048, 1, 187, "gru"),
std::make_tuple(2048, 2, 1500, "gru"),
std::make_tuple(2048, 2, 750, "gru"),
std::make_tuple(2048, 2, 375, "gru"),
std::make_tuple(2048, 2, 187, "gru"),
std::make_tuple(2048, 4, 1500, "gru"),
std::make_tuple(2048, 4, 750, "gru"),
std::make_tuple(2048, 4, 375, "gru"),
std::make_tuple(2048, 4, 187, "gru"),
std::make_tuple(1536, 1, 1500, "gru"),
std::make_tuple(1536, 1, 750, "gru"),
std::make_tuple(1536, 1, 375, "gru"),
std::make_tuple(1536, 1, 187, "gru"),
std::make_tuple(1536, 2, 1500, "gru"),
std::make_tuple(1536, 2, 750, "gru"),
std::make_tuple(1536, 2, 375, "gru"),
std::make_tuple(1536, 2, 187, "gru"),
std::make_tuple(1536, 4, 1500, "gru"),
std::make_tuple(1536, 4, 750, "gru"),
std::make_tuple(1536, 4, 375, "gru"),
std::make_tuple(1536, 4, 187, "gru"),
std::make_tuple(2560, 1, 1500, "gru"),
std::make_tuple(2560, 1, 750, "gru"),
std::make_tuple(2560, 1, 375, "gru"),
std::make_tuple(2560, 1, 187, "gru"),
std::make_tuple(2560, 2, 1500, "gru"),
std::make_tuple(2560, 2, 750, "gru"),
std::make_tuple(2560, 2, 375, "gru"),
std::make_tuple(2560, 2, 187, "gru"),
std::make_tuple(2560, 4, 1500, "gru"),
std::make_tuple(2560, 4, 750, "gru"),
std::make_tuple(2560, 4, 375, "gru"),
std::make_tuple(2560, 4, 187, "gru"),
std::make_tuple(512, 1, 1, "gru"),
std::make_tuple(512, 2, 1, "gru"),
std::make_tuple(512, 4, 1, "gru"),
std::make_tuple(1024, 1, 1500, "gru"),
std::make_tuple(1024, 2, 1500, "gru"),
std::make_tuple(1024, 4, 1500, "gru")
};
// Hidden units, batch size, timesteps, RNN cell type
std::vector<std::tuple<int, int, int, std::string>> inference_device_set = {
std::make_tuple(32, 1, 672, "vanilla"),
std::make_tuple(64, 1, 96, "vanilla"),
std::make_tuple(256, 1, 150, "lstm"),
};