Hugging Face's logo

major-scale
/
anim-binary-counting

world-model
dreamerv3
binary-arithmetic
mechanistic-interpretability
Model card Files
xet
 Community
major-scale commited on
Commit
57529d1
·
verified ·
1 Parent(s): 39b838d

Add weight manifest

Browse files
Files changed (1) hide show
  1. exported/dreamer_manifest.json +277 -0
exported/dreamer_manifest.json ADDED
@@ -0,0 +1,277 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tensors": {
3
+ "deter_init_w": {
4
+ "offset": 0,
5
+ "length": 512,
6
+ "shape": [
7
+ 1,
8
+ 512
9
+ ]
10
+ },
11
+ "gru_norm_b": {
12
+ "offset": 6293504,
13
+ "length": 1536,
14
+ "shape": [
15
+ 1536
16
+ ]
17
+ },
18
+ "gru_norm_w": {
19
+ "offset": 6299648,
20
+ "length": 1536,
21
+ "shape": [
22
+ 1536
23
+ ]
24
+ },
25
+ "img_in_w": {
26
+ "offset": 6305792,
27
+ "length": 524800,
28
+ "shape": [
29
+ 512,
30
+ 1025
31
+ ]
32
+ },
33
+ "img_in_norm_b": {
34
+ "offset": 8404992,
35
+ "length": 512,
36
+ "shape": [
37
+ 512
38
+ ]
39
+ },
40
+ "img_in_norm_w": {
41
+ "offset": 8407040,
42
+ "length": 512,
43
+ "shape": [
44
+ 512
45
+ ]
46
+ },
47
+ "img_out_w": {
48
+ "offset": 8409088,
49
+ "length": 262144,
50
+ "shape": [
51
+ 512,
52
+ 512
53
+ ]
54
+ },
55
+ "img_out_norm_b": {
56
+ "offset": 9457664,
57
+ "length": 512,
58
+ "shape": [
59
+ 512
60
+ ]
61
+ },
62
+ "img_out_norm_w": {
63
+ "offset": 9459712,
64
+ "length": 512,
65
+ "shape": [
66
+ 512
67
+ ]
68
+ },
69
+ "imgs_stat_b": {
70
+ "offset": 9461760,
71
+ "length": 1024,
72
+ "shape": [
73
+ 1024
74
+ ]
75
+ },
76
+ "imgs_stat_w": {
77
+ "offset": 9465856,
78
+ "length": 524288,
79
+ "shape": [
80
+ 1024,
81
+ 512
82
+ ]
83
+ },
84
+ "obs_out_w": {
85
+ "offset": 11563008,
86
+ "length": 524288,
87
+ "shape": [
88
+ 512,
89
+ 1024
90
+ ]
91
+ },
92
+ "obs_out_norm_b": {
93
+ "offset": 13660160,
94
+ "length": 512,
95
+ "shape": [
96
+ 512
97
+ ]
98
+ },
99
+ "obs_out_norm_w": {
100
+ "offset": 13662208,
101
+ "length": 512,
102
+ "shape": [
103
+ 512
104
+ ]
105
+ },
106
+ "obs_stat_b": {
107
+ "offset": 13664256,
108
+ "length": 1024,
109
+ "shape": [
110
+ 1024
111
+ ]
112
+ },
113
+ "obs_stat_w": {
114
+ "offset": 13668352,
115
+ "length": 524288,
116
+ "shape": [
117
+ 1024,
118
+ 512
119
+ ]
120
+ },
121
+ "enc_linear0_w": {
122
+ "offset": 15765504,
123
+ "length": 36864,
124
+ "shape": [
125
+ 512,
126
+ 72
127
+ ]
128
+ },
129
+ "enc_linear1_w": {
130
+ "offset": 15912960,
131
+ "length": 262144,
132
+ "shape": [
133
+ 512,
134
+ 512
135
+ ]
136
+ },
137
+ "enc_linear2_w": {
138
+ "offset": 16961536,
139
+ "length": 262144,
140
+ "shape": [
141
+ 512,
142
+ 512
143
+ ]
144
+ },
145
+ "enc_norm0_b": {
146
+ "offset": 18010112,
147
+ "length": 512,
148
+ "shape": [
149
+ 512
150
+ ]
151
+ },
152
+ "enc_norm0_w": {
153
+ "offset": 18012160,
154
+ "length": 512,
155
+ "shape": [
156
+ 512
157
+ ]
158
+ },
159
+ "enc_norm1_b": {
160
+ "offset": 18014208,
161
+ "length": 512,
162
+ "shape": [
163
+ 512
164
+ ]
165
+ },
166
+ "enc_norm1_w": {
167
+ "offset": 18016256,
168
+ "length": 512,
169
+ "shape": [
170
+ 512
171
+ ]
172
+ },
173
+ "enc_norm2_b": {
174
+ "offset": 18018304,
175
+ "length": 512,
176
+ "shape": [
177
+ 512
178
+ ]
179
+ },
180
+ "enc_norm2_w": {
181
+ "offset": 18020352,
182
+ "length": 512,
183
+ "shape": [
184
+ 512
185
+ ]
186
+ },
187
+ "dec_linear0_w": {
188
+ "offset": 18022400,
189
+ "length": 786432,
190
+ "shape": [
191
+ 512,
192
+ 1536
193
+ ]
194
+ },
195
+ "dec_linear1_w": {
196
+ "offset": 21168128,
197
+ "length": 262144,
198
+ "shape": [
199
+ 512,
200
+ 512
201
+ ]
202
+ },
203
+ "dec_linear2_w": {
204
+ "offset": 22216704,
205
+ "length": 262144,
206
+ "shape": [
207
+ 512,
208
+ 512
209
+ ]
210
+ },
211
+ "dec_norm0_b": {
212
+ "offset": 23265280,
213
+ "length": 512,
214
+ "shape": [
215
+ 512
216
+ ]
217
+ },
218
+ "dec_norm0_w": {
219
+ "offset": 23267328,
220
+ "length": 512,
221
+ "shape": [
222
+ 512
223
+ ]
224
+ },
225
+ "dec_norm1_b": {
226
+ "offset": 23269376,
227
+ "length": 512,
228
+ "shape": [
229
+ 512
230
+ ]
231
+ },
232
+ "dec_norm1_w": {
233
+ "offset": 23271424,
234
+ "length": 512,
235
+ "shape": [
236
+ 512
237
+ ]
238
+ },
239
+ "dec_norm2_b": {
240
+ "offset": 23273472,
241
+ "length": 512,
242
+ "shape": [
243
+ 512
244
+ ]
245
+ },
246
+ "dec_norm2_w": {
247
+ "offset": 23275520,
248
+ "length": 512,
249
+ "shape": [
250
+ 512
251
+ ]
252
+ },
253
+ "dec_out_b": {
254
+ "offset": 23277568,
255
+ "length": 72,
256
+ "shape": [
257
+ 72
258
+ ]
259
+ },
260
+ "dec_out_w": {
261
+ "offset": 23277856,
262
+ "length": 36864,
263
+ "shape": [
264
+ 72,
265
+ 512
266
+ ]
267
+ },
268
+ "gru_w": {
269
+ "offset": 2048,
270
+ "length": 1572864,
271
+ "shape": [
272
+ 1536,
273
+ 1024
274
+ ]
275
+ }
276
+ }
277
+ }