Reza2kn commited on
Commit
17efd9b
·
1 Parent(s): afeb07f

Release Gooya RizehPizeh v2 trainable checkpoint

Browse files
LICENSE ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ MIT License
2
+
3
+ Copyright (c) 2026 Reza2kn
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
README.md ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - fa
4
+ license: mit
5
+ pipeline_tag: text-to-speech
6
+ base_model: Reza2kn/Gooya-RizehPizeh-v1.5
7
+ library_name: piper
8
+ model_type: text-to-speech
9
+ tags:
10
+ - audio
11
+ - text-to-speech
12
+ - piper
13
+ - vits
14
+ - persian
15
+ - farsi
16
+ - negara
17
+ pretty_name: Gooya RizehPizeh v2
18
+ ---
19
+
20
+ # Gooya RizehPizeh v2
21
+
22
+ Persian (Farsi) single-speaker Piper/VITS2 voice fine-tuned on approved Gemini TTS Wikipedia audio. The model uses a case-sensitive Negara-style phonetic/Finglish frontend and produces 22,050 Hz mono audio.
23
+
24
+ ## Selected checkpoint
25
+
26
+ This release selects the checkpoint at **global step 50,000** (`val_mel = 0.3951`). Training continued to 100,000 global steps, but later validation was worse, so the terminal state is intentionally not promoted.
27
+
28
+ - Generator parameters: approximately 23.7M
29
+ - Full trainable checkpoint: `checkpoint/gooya-rizehpizeh-v2-step50000-val_mel0.3951.ckpt`
30
+ - Checkpoint SHA-256: `0c2d22718894e20fb6361c7bf2d75ee54d61db3a167983ce5d843ff6ed438938`
31
+ - Sample rate: 22,050 Hz
32
+ - Speakers: 1
33
+ - Phoneme type: `text`
34
+ - Symbol inventory: 256 IDs
35
+
36
+ ## Model lineage
37
+
38
+ The original v1.5 production ONNX did not have a matching resumable checkpoint. A trainable hybrid checkpoint was reconstructed by copying all 350 inference-time generator tensors from the production ONNX into a compatible Mana Piper training checkpoint. Training-only branches absent from ONNX came from that Mana template.
39
+
40
+ Before fine-tuning, a fresh ONNX round-trip matched the production model on 19/19 deterministic test rows:
41
+
42
+ - Worst maximum absolute error: `1.0348856449127197e-05`
43
+ - Minimum SNR: `110.0768 dB`
44
+ - Minimum correlation: `0.999999999995`
45
+
46
+ The complete recovery and parity receipts are under `training/`.
47
+
48
+ ## Fine-tuning data
49
+
50
+ The parent corpus came from approved rows in [`Reza2kn/Wikipedia-FA-EN-DeepSeek-V4-Flash-0731`](https://huggingface.co/datasets/Reza2kn/Wikipedia-FA-EN-DeepSeek-V4-Flash-0731).
51
+
52
+ - Requested parent clips: 4,186 (`3,517` train, `669` sealed eval)
53
+ - Successfully aligned parents: 4,116
54
+ - Rejected during alignment: 70
55
+ - Materialized segments: 30,965
56
+ - Training segments consumed: 26,053
57
+ - Sealed eval segments, not consumed by training: 4,912
58
+ - Total materialized duration: 68.9853 hours
59
+ - Segment duration: mean `8.0203s`, min `0.6307s`, max `21.6656s`
60
+
61
+ Audio timing came from AvaSanj v1 CTC forced alignment. Training phoneme labels came from Negara v7.1. Parent-level train/eval boundaries were preserved.
62
+
63
+ ## Training
64
+
65
+ - Precision: BF16 mixed precision
66
+ - Generator learning rate: `1e-5`
67
+ - Discriminator learning rate: `5e-6`
68
+ - Batch size: 1
69
+ - Selected validation metric: `val_mel`
70
+ - Selected checkpoint: global step 50,000, `val_mel = 0.3951`
71
+
72
+ The 100K run was resumed from a successful 2K signal checkpoint. The sealed 669-parent eval split was not used for checkpoint selection.
73
+
74
+ ## Inference and long-form use
75
+
76
+ Input must already be case-sensitive Negara-compatible phonetic text. Example:
77
+
78
+ ```bash
79
+ echo "zurxAneye vahdat yek makAne varzeSi dar Sahre borujerd ast." | \
80
+ piper --model gooya-rizehpizeh-v2.onnx \
81
+ --config gooya-rizehpizeh-v2.onnx.json \
82
+ --output_file output.wav
83
+ ```
84
+
85
+ For long-form narration, do not send an article as one model input. Split at semantic sentence/clause boundaries into substantial spans, synthesize each span independently, trim quiet waveform edges, and join with boundary-aware pauses. The final output duration can be arbitrarily long even though each model invocation remains bounded.
86
+
87
+ ## Limitations
88
+
89
+ - Lower validation mel loss did not guarantee universally better pronunciation. Human listening found some words, including proper nouns, better at the earlier 2K checkpoint or the original v1.5 model.
90
+ - G2P/Finglish quality remains the dominant source of pronunciation errors.
91
+ - Very long single-pass phoneme sequences can collapse or omit content; use chunked synthesis.
92
+ - Overly small chunks sound breathy and robotic. Prefer substantial semantic spans instead of fixed tiny fragments.
93
+ - This release has not yet been promoted over v1.5 on Gouya.app.
94
+
95
+ ## License
96
+
97
+ MIT. See `LICENSE`.
checkpoint/gooya-rizehpizeh-v2-step50000-val_mel0.3951.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c2d22718894e20fb6361c7bf2d75ee54d61db3a167983ce5d843ff6ed438938
3
+ size 845890322
config/negara-phoneme-id-map.json ADDED
@@ -0,0 +1,498 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ " ": [
3
+ 3
4
+ ],
5
+ "!": [
6
+ 4
7
+ ],
8
+ "\"": [
9
+ 150
10
+ ],
11
+ "#": [
12
+ 149
13
+ ],
14
+ "$": [
15
+ 2
16
+ ],
17
+ "'": [
18
+ 5
19
+ ],
20
+ "(": [
21
+ 6
22
+ ],
23
+ ")": [
24
+ 7
25
+ ],
26
+ ",": [
27
+ 8
28
+ ],
29
+ "-": [
30
+ 9
31
+ ],
32
+ ".": [
33
+ 10
34
+ ],
35
+ "0": [
36
+ 130
37
+ ],
38
+ "1": [
39
+ 131
40
+ ],
41
+ "2": [
42
+ 132
43
+ ],
44
+ "3": [
45
+ 133
46
+ ],
47
+ "4": [
48
+ 134
49
+ ],
50
+ "5": [
51
+ 135
52
+ ],
53
+ "6": [
54
+ 136
55
+ ],
56
+ "7": [
57
+ 137
58
+ ],
59
+ "8": [
60
+ 138
61
+ ],
62
+ "9": [
63
+ 139
64
+ ],
65
+ ":": [
66
+ 11
67
+ ],
68
+ ";": [
69
+ 12
70
+ ],
71
+ "?": [
72
+ 109
73
+ ],
74
+ "A": [
75
+ 52,
76
+ 0,
77
+ 122
78
+ ],
79
+ "C": [
80
+ 32,
81
+ 0,
82
+ 96
83
+ ],
84
+ "Q": [
85
+ 109
86
+ ],
87
+ "S": [
88
+ 96
89
+ ],
90
+ "X": [
91
+ 156
92
+ ],
93
+ "Z": [
94
+ 108
95
+ ],
96
+ "^": [
97
+ 1
98
+ ],
99
+ "_": [
100
+ 0
101
+ ],
102
+ "a": [
103
+ 39
104
+ ],
105
+ "b": [
106
+ 15
107
+ ],
108
+ "c": [
109
+ 16
110
+ ],
111
+ "d": [
112
+ 17
113
+ ],
114
+ "e": [
115
+ 18
116
+ ],
117
+ "f": [
118
+ 19
119
+ ],
120
+ "g": [
121
+ 66
122
+ ],
123
+ "h": [
124
+ 20
125
+ ],
126
+ "i": [
127
+ 21,
128
+ 0,
129
+ 122
130
+ ],
131
+ "j": [
132
+ 17,
133
+ 0,
134
+ 108
135
+ ],
136
+ "k": [
137
+ 23
138
+ ],
139
+ "l": [
140
+ 24
141
+ ],
142
+ "m": [
143
+ 25
144
+ ],
145
+ "n": [
146
+ 26
147
+ ],
148
+ "o": [
149
+ 27
150
+ ],
151
+ "p": [
152
+ 28
153
+ ],
154
+ "q": [
155
+ 68
156
+ ],
157
+ "r": [
158
+ 92
159
+ ],
160
+ "s": [
161
+ 31
162
+ ],
163
+ "t": [
164
+ 32
165
+ ],
166
+ "u": [
167
+ 33,
168
+ 0,
169
+ 122
170
+ ],
171
+ "v": [
172
+ 34
173
+ ],
174
+ "w": [
175
+ 35
176
+ ],
177
+ "x": [
178
+ 36
179
+ ],
180
+ "y": [
181
+ 22
182
+ ],
183
+ "z": [
184
+ 38
185
+ ],
186
+ "æ": [
187
+ 39
188
+ ],
189
+ "ç": [
190
+ 40
191
+ ],
192
+ "ð": [
193
+ 41
194
+ ],
195
+ "ø": [
196
+ 42
197
+ ],
198
+ "ħ": [
199
+ 43
200
+ ],
201
+ "ŋ": [
202
+ 44
203
+ ],
204
+ "œ": [
205
+ 45
206
+ ],
207
+ "ǀ": [
208
+ 46
209
+ ],
210
+ "ǁ": [
211
+ 47
212
+ ],
213
+ "ǂ": [
214
+ 48
215
+ ],
216
+ "ǃ": [
217
+ 49
218
+ ],
219
+ "ɐ": [
220
+ 50
221
+ ],
222
+ "ɑ": [
223
+ 51
224
+ ],
225
+ "ɒ": [
226
+ 52
227
+ ],
228
+ "ɓ": [
229
+ 53
230
+ ],
231
+ "ɔ": [
232
+ 54
233
+ ],
234
+ "ɕ": [
235
+ 55
236
+ ],
237
+ "ɖ": [
238
+ 56
239
+ ],
240
+ "ɗ": [
241
+ 57
242
+ ],
243
+ "ɘ": [
244
+ 58
245
+ ],
246
+ "ə": [
247
+ 59
248
+ ],
249
+ "ɚ": [
250
+ 60
251
+ ],
252
+ "ɛ": [
253
+ 61
254
+ ],
255
+ "ɜ": [
256
+ 62
257
+ ],
258
+ "ɞ": [
259
+ 63
260
+ ],
261
+ "ɟ": [
262
+ 64
263
+ ],
264
+ "ɠ": [
265
+ 65
266
+ ],
267
+ "ɡ": [
268
+ 66
269
+ ],
270
+ "ɢ": [
271
+ 67
272
+ ],
273
+ "ɣ": [
274
+ 68
275
+ ],
276
+ "ɤ": [
277
+ 69
278
+ ],
279
+ "ɥ": [
280
+ 70
281
+ ],
282
+ "ɦ": [
283
+ 71
284
+ ],
285
+ "ɧ": [
286
+ 72
287
+ ],
288
+ "ɨ": [
289
+ 73
290
+ ],
291
+ "ɪ": [
292
+ 74
293
+ ],
294
+ "ɫ": [
295
+ 75
296
+ ],
297
+ "ɬ": [
298
+ 76
299
+ ],
300
+ "ɭ": [
301
+ 77
302
+ ],
303
+ "ɮ": [
304
+ 78
305
+ ],
306
+ "ɯ": [
307
+ 79
308
+ ],
309
+ "ɰ": [
310
+ 80
311
+ ],
312
+ "ɱ": [
313
+ 81
314
+ ],
315
+ "ɲ": [
316
+ 82
317
+ ],
318
+ "ɳ": [
319
+ 83
320
+ ],
321
+ "ɴ": [
322
+ 84
323
+ ],
324
+ "ɵ": [
325
+ 85
326
+ ],
327
+ "ɶ": [
328
+ 86
329
+ ],
330
+ "ɸ": [
331
+ 87
332
+ ],
333
+ "ɹ": [
334
+ 88
335
+ ],
336
+ "ɺ": [
337
+ 89
338
+ ],
339
+ "ɻ": [
340
+ 90
341
+ ],
342
+ "ɽ": [
343
+ 91
344
+ ],
345
+ "ɾ": [
346
+ 92
347
+ ],
348
+ "ʀ": [
349
+ 93
350
+ ],
351
+ "ʁ": [
352
+ 94
353
+ ],
354
+ "ʂ": [
355
+ 95
356
+ ],
357
+ "ʃ": [
358
+ 96
359
+ ],
360
+ "ʄ": [
361
+ 97
362
+ ],
363
+ "ʈ": [
364
+ 98
365
+ ],
366
+ "ʉ": [
367
+ 99
368
+ ],
369
+ "ʊ": [
370
+ 100
371
+ ],
372
+ "ʋ": [
373
+ 101
374
+ ],
375
+ "ʌ": [
376
+ 102
377
+ ],
378
+ "ʍ": [
379
+ 103
380
+ ],
381
+ "ʎ": [
382
+ 104
383
+ ],
384
+ "ʏ": [
385
+ 105
386
+ ],
387
+ "ʐ": [
388
+ 106
389
+ ],
390
+ "ʑ": [
391
+ 107
392
+ ],
393
+ "ʒ": [
394
+ 108
395
+ ],
396
+ "ʔ": [
397
+ 109
398
+ ],
399
+ "ʕ": [
400
+ 110
401
+ ],
402
+ "ʘ": [
403
+ 111
404
+ ],
405
+ "ʙ": [
406
+ 112
407
+ ],
408
+ "ʛ": [
409
+ 113
410
+ ],
411
+ "ʜ": [
412
+ 114
413
+ ],
414
+ "ʝ": [
415
+ 115
416
+ ],
417
+ "ʟ": [
418
+ 116
419
+ ],
420
+ "ʡ": [
421
+ 117
422
+ ],
423
+ "ʢ": [
424
+ 118
425
+ ],
426
+ "ʦ": [
427
+ 155
428
+ ],
429
+ "ʰ": [
430
+ 145
431
+ ],
432
+ "ʲ": [
433
+ 119
434
+ ],
435
+ "ˈ": [
436
+ 120
437
+ ],
438
+ "ˌ": [
439
+ 121
440
+ ],
441
+ "ː": [
442
+ 122
443
+ ],
444
+ "ˑ": [
445
+ 123
446
+ ],
447
+ "˞": [
448
+ 124
449
+ ],
450
+ "ˤ": [
451
+ 146
452
+ ],
453
+ "̃": [
454
+ 141
455
+ ],
456
+ "̧": [
457
+ 140
458
+ ],
459
+ "̩": [
460
+ 144
461
+ ],
462
+ "̪": [
463
+ 142
464
+ ],
465
+ "̯": [
466
+ 143
467
+ ],
468
+ "̺": [
469
+ 152
470
+ ],
471
+ "̻": [
472
+ 153
473
+ ],
474
+ "β": [
475
+ 125
476
+ ],
477
+ "ε": [
478
+ 147
479
+ ],
480
+ "θ": [
481
+ 126
482
+ ],
483
+ "χ": [
484
+ 127
485
+ ],
486
+ "ᵻ": [
487
+ 128
488
+ ],
489
+ "↑": [
490
+ 151
491
+ ],
492
+ "↓": [
493
+ 148
494
+ ],
495
+ "ⱱ": [
496
+ 129
497
+ ]
498
+ }
release-manifest.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "files": [
3
+ {
4
+ "bytes": 1519,
5
+ "path": ".gitattributes",
6
+ "sha256": "11ad7efa24975ee4b0c3c3a38ed18737f0658a5f75a0a96787b576a78a023361"
7
+ },
8
+ {
9
+ "bytes": 1064,
10
+ "path": "LICENSE",
11
+ "sha256": "beacc839712cc8db59ad2997b52edb77946461593219d82276e39ba7e404f7ef"
12
+ },
13
+ {
14
+ "bytes": 4057,
15
+ "path": "README.md",
16
+ "sha256": "997ae579d683bead1b159189bf598202a4fdee03122c90132d06857f823d443e"
17
+ },
18
+ {
19
+ "bytes": 845890322,
20
+ "path": "checkpoint/gooya-rizehpizeh-v2-step50000-val_mel0.3951.ckpt",
21
+ "sha256": "0c2d22718894e20fb6361c7bf2d75ee54d61db3a167983ce5d843ff6ed438938"
22
+ },
23
+ {
24
+ "bytes": 3642,
25
+ "path": "config/negara-phoneme-id-map.json",
26
+ "sha256": "6e67148a46b9bf5a3529c97cc1fa60a10f477d0bd80e61a1fb1a24ad59eefe83"
27
+ },
28
+ {
29
+ "bytes": 1903,
30
+ "path": "training/alignment-report.json",
31
+ "sha256": "b8dff1be540f03404b05d552eac009f5ca1781c5190898622851331ac44de325"
32
+ },
33
+ {
34
+ "bytes": 9626,
35
+ "path": "training/base-onnx-roundtrip-parity.json",
36
+ "sha256": "1e1346518a1cd72c7b8d9368845c89206ba026d78417af753a32f9c41fdaef59"
37
+ },
38
+ {
39
+ "bytes": 82458,
40
+ "path": "training/checkpoint-recovery-report.json",
41
+ "sha256": "ec0bb47bbf07c5bfeff92ab68ec57e1099d94cd5172d087f8f6694e5ce20e178"
42
+ },
43
+ {
44
+ "bytes": 2913,
45
+ "path": "training/train_rizeh_full_100k.sh",
46
+ "sha256": "3b7a08aee6af72a787000a62b209fdf70c8726a65a8a5e66176c4106158f392c"
47
+ }
48
+ ],
49
+ "schema_version": "gooya.rizehpizeh-v2-release/v1",
50
+ "selected_global_step": 50000,
51
+ "selected_val_mel": 0.3951,
52
+ "status": "trainable-checkpoint-ready"
53
+ }
training/alignment-report.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "models": {
3
+ "avasanj_model_sha256": "c7ea4f6c1e510e4c23bbc1856c290787f20a730d14c4d95de45c03259d5dd07c",
4
+ "mana_config_sha256": "df13724ca155a8699d8c42d47d5748dee98ec8057cafdaed8b1455fc85798753",
5
+ "mana_phoneme_map_sha256": "ba38263aec458bca093a0e60629a442dfb365bd7ef79e70be58c311ba886f868",
6
+ "negara_model_sha256": "2d89883abe07082cfb291203c1e5609f8974e71ca04d6c3e73bf2d32f8b715cf",
7
+ "negara_overlay_sha256": "9e063c2f26db4f56f6a93fef33f1f2a25423318ff7073b9954fddf4c4085412f",
8
+ "negara_runtime_map_sha256": "6e67148a46b9bf5a3529c97cc1fa60a10f477d0bd80e61a1fb1a24ad59eefe83"
9
+ },
10
+ "outputs": {
11
+ "eval_metadata_sha256": "587ff471a78e61f471d524435f9bbe9c750378861185bf27b88656d27d41652d",
12
+ "segments_sha256": "f7fd2abb87bd73b4e80dc8a5cd23acb9a8fafd480d56ab9d500bacbbb7b5d4ed",
13
+ "train_metadata_sha256": "3077e23e86a97454b36c1d5c7d2b34108b332752b3f600811418714af8928320"
14
+ },
15
+ "parent_manifest": "/home/rezo/gooya-bozorg-v15-gemini-approved-20260829/training_manifest.jsonl",
16
+ "parent_manifest_sha256": "d285458025385945d10e733bdea74080f544c4aa3c621454686d6ffe7678f345",
17
+ "parent_rows_requested": 4186,
18
+ "parent_split_counts": {
19
+ "eval": 669,
20
+ "train": 3517
21
+ },
22
+ "parent_status_counts": {
23
+ "complete": 4116,
24
+ "rejected": 70
25
+ },
26
+ "schema_version": "gooya.piper-wikipedia-aligned/v1",
27
+ "segment_duration_seconds": {
28
+ "max": 21.665583333333334,
29
+ "mean": 8.020255083696643,
30
+ "min": 0.6307083333333333
31
+ },
32
+ "segment_hours": 68.98533296296293,
33
+ "segment_rows": 30965,
34
+ "segment_split_counts": {
35
+ "eval": 4912,
36
+ "train": 26053
37
+ },
38
+ "segmentation": {
39
+ "audio_timing_source": "AvaSanj v1 CTC forced alignment",
40
+ "edge_padding_seconds": 0.08,
41
+ "maximum_seconds": 12.0,
42
+ "minimum_seconds": 2.0,
43
+ "parent_split_preserved": true,
44
+ "target_seconds": 8.0,
45
+ "training_label_source": "Negara v7.1"
46
+ },
47
+ "status": "complete"
48
+ }
training/base-onnx-roundtrip-parity.json ADDED
@@ -0,0 +1,256 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "deterministic_scales": [
3
+ 0.0,
4
+ 1.0,
5
+ 0.0
6
+ ],
7
+ "production_onnx": "/home/rezo/piper-wikipedia-approved-20260830/models/rizeh-onnx-recovery/epoch4-rizehpizeh.onnx",
8
+ "production_onnx_sha256": "dc8532f0c57a40ed076e466f380376577e8bbcd13d1fc19c81f69d461d26dadc",
9
+ "roundtrip_onnx": "/home/rezo/piper-wikipedia-approved-20260830/models/rizeh-onnx-recovery/rizehpizeh-epoch4-roundtrip.onnx",
10
+ "roundtrip_onnx_sha256": "f676eb48c3671dc67a57f21fa64778d66bbdd7c6af7dc9225f93a90160118f46",
11
+ "rows": [
12
+ {
13
+ "accepted": true,
14
+ "correlation": 0.9999999999950887,
15
+ "lengths_equal": true,
16
+ "maximum_absolute_error": 8.016824722290039e-06,
17
+ "phone_ids": 415,
18
+ "production_samples": 129024,
19
+ "rmse": 2.5333728495926234e-07,
20
+ "roundtrip_samples": 129024,
21
+ "snr_db": 110.07684111073353,
22
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0002-46a7c32056eb--000.flac"
23
+ },
24
+ {
25
+ "accepted": true,
26
+ "correlation": 0.9999999999976001,
27
+ "lengths_equal": true,
28
+ "maximum_absolute_error": 6.645917892456055e-06,
29
+ "phone_ids": 409,
30
+ "production_samples": 167680,
31
+ "rmse": 1.9315625937858818e-07,
32
+ "roundtrip_samples": 167680,
33
+ "snr_db": 113.1891144903417,
34
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0002-46a7c32056eb--001.flac"
35
+ },
36
+ {
37
+ "accepted": true,
38
+ "correlation": 0.9999999999992383,
39
+ "lengths_equal": true,
40
+ "maximum_absolute_error": 4.634261131286621e-06,
41
+ "phone_ids": 369,
42
+ "production_samples": 147200,
43
+ "rmse": 1.0709432543531402e-07,
44
+ "roundtrip_samples": 147200,
45
+ "snr_db": 118.17123906562294,
46
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0002-46a7c32056eb--002.flac"
47
+ },
48
+ {
49
+ "accepted": true,
50
+ "correlation": 0.9999999999996858,
51
+ "lengths_equal": true,
52
+ "maximum_absolute_error": 1.866370439529419e-06,
53
+ "phone_ids": 303,
54
+ "production_samples": 135424,
55
+ "rmse": 7.979961651606776e-08,
56
+ "roundtrip_samples": 135424,
57
+ "snr_db": 122.00237143262127,
58
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0002-46a7c32056eb--003.flac"
59
+ },
60
+ {
61
+ "accepted": true,
62
+ "correlation": 0.9999999999989482,
63
+ "lengths_equal": true,
64
+ "maximum_absolute_error": 3.6954879760742188e-06,
65
+ "phone_ids": 383,
66
+ "production_samples": 154368,
67
+ "rmse": 1.3961939242212696e-07,
68
+ "roundtrip_samples": 154368,
69
+ "snr_db": 116.75999407976957,
70
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0002-46a7c32056eb--004.flac"
71
+ },
72
+ {
73
+ "accepted": true,
74
+ "correlation": 0.9999999999975221,
75
+ "lengths_equal": true,
76
+ "maximum_absolute_error": 1.0348856449127197e-05,
77
+ "phone_ids": 353,
78
+ "production_samples": 148224,
79
+ "rmse": 1.9280742309236266e-07,
80
+ "roundtrip_samples": 148224,
81
+ "snr_db": 113.04745851770522,
82
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0005-7fdfa3438fa0--000.flac"
83
+ },
84
+ {
85
+ "accepted": true,
86
+ "correlation": 0.9999999999990786,
87
+ "lengths_equal": true,
88
+ "maximum_absolute_error": 3.337860107421875e-06,
89
+ "phone_ids": 315,
90
+ "production_samples": 142336,
91
+ "rmse": 1.292967613370121e-07,
92
+ "roundtrip_samples": 142336,
93
+ "snr_db": 117.34298666422276,
94
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0005-7fdfa3438fa0--001.flac"
95
+ },
96
+ {
97
+ "accepted": true,
98
+ "correlation": 0.9999999999988867,
99
+ "lengths_equal": true,
100
+ "maximum_absolute_error": 5.193054676055908e-06,
101
+ "phone_ids": 349,
102
+ "production_samples": 167936,
103
+ "rmse": 1.4744958703662086e-07,
104
+ "roundtrip_samples": 167936,
105
+ "snr_db": 116.51689647397932,
106
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0005-7fdfa3438fa0--002.flac"
107
+ },
108
+ {
109
+ "accepted": true,
110
+ "correlation": 0.9999999999990954,
111
+ "lengths_equal": true,
112
+ "maximum_absolute_error": 3.28943133354187e-06,
113
+ "phone_ids": 297,
114
+ "production_samples": 128256,
115
+ "rmse": 1.303211847859652e-07,
116
+ "roundtrip_samples": 128256,
117
+ "snr_db": 117.41796567668571,
118
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0005-7fdfa3438fa0--003.flac"
119
+ },
120
+ {
121
+ "accepted": true,
122
+ "correlation": 0.9999999999985223,
123
+ "lengths_equal": true,
124
+ "maximum_absolute_error": 3.516674041748047e-06,
125
+ "phone_ids": 313,
126
+ "production_samples": 143872,
127
+ "rmse": 1.6290985783034047e-07,
128
+ "roundtrip_samples": 143872,
129
+ "snr_db": 115.29210821485962,
130
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0005-7fdfa3438fa0--004.flac"
131
+ },
132
+ {
133
+ "accepted": true,
134
+ "correlation": 0.9999999999983282,
135
+ "lengths_equal": true,
136
+ "maximum_absolute_error": 6.0908496379852295e-06,
137
+ "phone_ids": 305,
138
+ "production_samples": 149760,
139
+ "rmse": 1.770678269572096e-07,
140
+ "roundtrip_samples": 149760,
141
+ "snr_db": 114.75974696484796,
142
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0005-7fdfa3438fa0--005.flac"
143
+ },
144
+ {
145
+ "accepted": true,
146
+ "correlation": 0.9999999999997442,
147
+ "lengths_equal": true,
148
+ "maximum_absolute_error": 1.0058283805847168e-06,
149
+ "phone_ids": 133,
150
+ "production_samples": 65280,
151
+ "rmse": 7.696038119357825e-08,
152
+ "roundtrip_samples": 65280,
153
+ "snr_db": 122.88017157660953,
154
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0005-7fdfa3438fa0--006.flac"
155
+ },
156
+ {
157
+ "accepted": true,
158
+ "correlation": 0.9999999999985759,
159
+ "lengths_equal": true,
160
+ "maximum_absolute_error": 5.036592483520508e-06,
161
+ "phone_ids": 291,
162
+ "production_samples": 119552,
163
+ "rmse": 1.674838109233354e-07,
164
+ "roundtrip_samples": 119552,
165
+ "snr_db": 115.45141939011005,
166
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0006-6a7e88311c1c--000.flac"
167
+ },
168
+ {
169
+ "accepted": true,
170
+ "correlation": 0.9999999999991662,
171
+ "lengths_equal": true,
172
+ "maximum_absolute_error": 2.3096799850463867e-06,
173
+ "phone_ids": 251,
174
+ "production_samples": 116992,
175
+ "rmse": 1.224142861371859e-07,
176
+ "roundtrip_samples": 116992,
177
+ "snr_db": 117.77634522449642,
178
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0006-6a7e88311c1c--001.flac"
179
+ },
180
+ {
181
+ "accepted": true,
182
+ "correlation": 0.9999999999988628,
183
+ "lengths_equal": true,
184
+ "maximum_absolute_error": 3.3080577850341797e-06,
185
+ "phone_ids": 323,
186
+ "production_samples": 142592,
187
+ "rmse": 1.3896321899242215e-07,
188
+ "roundtrip_samples": 142592,
189
+ "snr_db": 116.43469302037973,
190
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0006-6a7e88311c1c--002.flac"
191
+ },
192
+ {
193
+ "accepted": true,
194
+ "correlation": 0.9999999999983126,
195
+ "lengths_equal": true,
196
+ "maximum_absolute_error": 5.781650543212891e-06,
197
+ "phone_ids": 259,
198
+ "production_samples": 117248,
199
+ "rmse": 1.8041357061800095e-07,
200
+ "roundtrip_samples": 117248,
201
+ "snr_db": 114.7142544343226,
202
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0006-6a7e88311c1c--003.flac"
203
+ },
204
+ {
205
+ "accepted": true,
206
+ "correlation": 0.9999999999987348,
207
+ "lengths_equal": true,
208
+ "maximum_absolute_error": 3.4123659133911133e-06,
209
+ "phone_ids": 305,
210
+ "production_samples": 133376,
211
+ "rmse": 1.4938583978210013e-07,
212
+ "roundtrip_samples": 133376,
213
+ "snr_db": 115.9668570899362,
214
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0006-6a7e88311c1c--004.flac"
215
+ },
216
+ {
217
+ "accepted": true,
218
+ "correlation": 0.9999999999979307,
219
+ "lengths_equal": true,
220
+ "maximum_absolute_error": 7.170019671320915e-06,
221
+ "phone_ids": 225,
222
+ "production_samples": 97024,
223
+ "rmse": 2.0489704485742987e-07,
224
+ "roundtrip_samples": 97024,
225
+ "snr_db": 113.8300351127441,
226
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0006-6a7e88311c1c--005.flac"
227
+ },
228
+ {
229
+ "accepted": true,
230
+ "correlation": 0.9999999999968524,
231
+ "lengths_equal": true,
232
+ "maximum_absolute_error": 4.556030035018921e-06,
233
+ "phone_ids": 207,
234
+ "production_samples": 91648,
235
+ "rmse": 2.3073153380373358e-07,
236
+ "roundtrip_samples": 91648,
237
+ "snr_db": 112.00926041459346,
238
+ "utterance": "/home/rezo/piper-wikipedia-approved-20260830/data/pilot3/audio/train/wiki-fa-0006-6a7e88311c1c--006.flac"
239
+ }
240
+ ],
241
+ "schema_version": "gooya.rizeh-onnx-roundtrip/v1",
242
+ "status": "accepted",
243
+ "summary": {
244
+ "accepted_rows": 19,
245
+ "length_matches": 19,
246
+ "minimum_correlation": 0.9999999999950887,
247
+ "minimum_snr_db": 110.07684111073353,
248
+ "rows": 19,
249
+ "worst_maximum_absolute_error": 1.0348856449127197e-05
250
+ },
251
+ "thresholds": {
252
+ "maximum_absolute_error": 2e-05,
253
+ "minimum_correlation": 0.99999999,
254
+ "minimum_snr_db": 90.0
255
+ }
256
+ }
training/checkpoint-recovery-report.json ADDED
@@ -0,0 +1,3479 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "inputs": {
3
+ "mapping_reference_checkpoint": "/home/rezo/piper-wikipedia-approved-20260830/runs/mana-warmstart-pilot3-runtimecheck/lightning_logs/version_0/checkpoints/last.ckpt",
4
+ "mapping_reference_checkpoint_sha256": "720e6f1deb9b665fa3a64b849578e67b22e5cd5457589492748c59b5af629239",
5
+ "mapping_reference_onnx": "/home/rezo/piper-wikipedia-approved-20260830/runs/mana-warmstart-pilot3-runtimecheck/model.onnx",
6
+ "mapping_reference_onnx_sha256": "b23dde224885d42fdddec8aa886e24fc8fa9cde7f188622e2771eb0ff2d4c2be",
7
+ "production_onnx": "/home/rezo/piper-wikipedia-approved-20260830/models/rizeh-onnx-recovery/epoch4-rizehpizeh.onnx",
8
+ "production_onnx_sha256": "dc8532f0c57a40ed076e466f380376577e8bbcd13d1fc19c81f69d461d26dadc",
9
+ "template_checkpoint": "/home/rezo/piper-wikipedia-approved-20260830/models/mana/ckpt/epoch=6012-step=4203520.ckpt",
10
+ "template_checkpoint_sha256": "ec2c289b8f4a32b0a7f1f491a98beaa24f8fab8a13de537495f1d649c8818980"
11
+ },
12
+ "output_checkpoint": "/home/rezo/piper-wikipedia-approved-20260830/models/rizeh-onnx-recovery/rizehpizeh-epoch4-recovered-trainable.ckpt",
13
+ "output_checkpoint_sha256": "331c1e8c95e86f88f2a11a6eb14475da609a58891a014cf3cf1e3af414210347",
14
+ "recovered_by_role": {
15
+ "dec": 45,
16
+ "dp": 114,
17
+ "enc_p": 111,
18
+ "flow": 80
19
+ },
20
+ "recovered_inference_tensors": 350,
21
+ "schema_version": "gooya.rizeh-onnx-recovery/v1",
22
+ "special_transform_verification": {
23
+ "initializer": "onnx::Exp_8946",
24
+ "max_abs_error": 0.0,
25
+ "transform": "checkpoint_value = -onnx_initializer"
26
+ },
27
+ "status": "recovered_pending_roundtrip_parity",
28
+ "tensors": [
29
+ {
30
+ "initializer": "sid",
31
+ "shape": [
32
+ 256,
33
+ 192
34
+ ],
35
+ "storage": "direct",
36
+ "target": "enc_p.emb.weight",
37
+ "transform": "identity"
38
+ },
39
+ {
40
+ "initializer": "enc_p.encoder.attn_layers.0.emb_rel_k",
41
+ "shape": [
42
+ 1,
43
+ 9,
44
+ 96
45
+ ],
46
+ "storage": "direct",
47
+ "target": "enc_p.encoder.attn_layers.0.emb_rel_k",
48
+ "transform": "identity"
49
+ },
50
+ {
51
+ "initializer": "enc_p.encoder.attn_layers.0.emb_rel_v",
52
+ "shape": [
53
+ 1,
54
+ 9,
55
+ 96
56
+ ],
57
+ "storage": "direct",
58
+ "target": "enc_p.encoder.attn_layers.0.emb_rel_v",
59
+ "transform": "identity"
60
+ },
61
+ {
62
+ "initializer": "enc_p.encoder.attn_layers.0.conv_q.weight",
63
+ "shape": [
64
+ 192,
65
+ 192,
66
+ 1
67
+ ],
68
+ "storage": "direct",
69
+ "target": "enc_p.encoder.attn_layers.0.conv_q.weight",
70
+ "transform": "identity"
71
+ },
72
+ {
73
+ "initializer": "enc_p.encoder.attn_layers.0.conv_q.bias",
74
+ "shape": [
75
+ 192
76
+ ],
77
+ "storage": "direct",
78
+ "target": "enc_p.encoder.attn_layers.0.conv_q.bias",
79
+ "transform": "identity"
80
+ },
81
+ {
82
+ "initializer": "enc_p.encoder.attn_layers.0.conv_k.weight",
83
+ "shape": [
84
+ 192,
85
+ 192,
86
+ 1
87
+ ],
88
+ "storage": "direct",
89
+ "target": "enc_p.encoder.attn_layers.0.conv_k.weight",
90
+ "transform": "identity"
91
+ },
92
+ {
93
+ "initializer": "enc_p.encoder.attn_layers.0.conv_k.bias",
94
+ "shape": [
95
+ 192
96
+ ],
97
+ "storage": "direct",
98
+ "target": "enc_p.encoder.attn_layers.0.conv_k.bias",
99
+ "transform": "identity"
100
+ },
101
+ {
102
+ "initializer": "enc_p.encoder.attn_layers.0.conv_v.weight",
103
+ "shape": [
104
+ 192,
105
+ 192,
106
+ 1
107
+ ],
108
+ "storage": "direct",
109
+ "target": "enc_p.encoder.attn_layers.0.conv_v.weight",
110
+ "transform": "identity"
111
+ },
112
+ {
113
+ "initializer": "enc_p.encoder.attn_layers.0.conv_v.bias",
114
+ "shape": [
115
+ 192
116
+ ],
117
+ "storage": "direct",
118
+ "target": "enc_p.encoder.attn_layers.0.conv_v.bias",
119
+ "transform": "identity"
120
+ },
121
+ {
122
+ "initializer": "enc_p.encoder.attn_layers.0.conv_o.weight",
123
+ "shape": [
124
+ 192,
125
+ 192,
126
+ 1
127
+ ],
128
+ "storage": "direct",
129
+ "target": "enc_p.encoder.attn_layers.0.conv_o.weight",
130
+ "transform": "identity"
131
+ },
132
+ {
133
+ "initializer": "enc_p.encoder.attn_layers.0.conv_o.bias",
134
+ "shape": [
135
+ 192
136
+ ],
137
+ "storage": "direct",
138
+ "target": "enc_p.encoder.attn_layers.0.conv_o.bias",
139
+ "transform": "identity"
140
+ },
141
+ {
142
+ "initializer": "enc_p.encoder.attn_layers.1.emb_rel_k",
143
+ "shape": [
144
+ 1,
145
+ 9,
146
+ 96
147
+ ],
148
+ "storage": "direct",
149
+ "target": "enc_p.encoder.attn_layers.1.emb_rel_k",
150
+ "transform": "identity"
151
+ },
152
+ {
153
+ "initializer": "enc_p.encoder.attn_layers.1.emb_rel_v",
154
+ "shape": [
155
+ 1,
156
+ 9,
157
+ 96
158
+ ],
159
+ "storage": "direct",
160
+ "target": "enc_p.encoder.attn_layers.1.emb_rel_v",
161
+ "transform": "identity"
162
+ },
163
+ {
164
+ "initializer": "enc_p.encoder.attn_layers.1.conv_q.weight",
165
+ "shape": [
166
+ 192,
167
+ 192,
168
+ 1
169
+ ],
170
+ "storage": "direct",
171
+ "target": "enc_p.encoder.attn_layers.1.conv_q.weight",
172
+ "transform": "identity"
173
+ },
174
+ {
175
+ "initializer": "enc_p.encoder.attn_layers.1.conv_q.bias",
176
+ "shape": [
177
+ 192
178
+ ],
179
+ "storage": "direct",
180
+ "target": "enc_p.encoder.attn_layers.1.conv_q.bias",
181
+ "transform": "identity"
182
+ },
183
+ {
184
+ "initializer": "enc_p.encoder.attn_layers.1.conv_k.weight",
185
+ "shape": [
186
+ 192,
187
+ 192,
188
+ 1
189
+ ],
190
+ "storage": "direct",
191
+ "target": "enc_p.encoder.attn_layers.1.conv_k.weight",
192
+ "transform": "identity"
193
+ },
194
+ {
195
+ "initializer": "enc_p.encoder.attn_layers.1.conv_k.bias",
196
+ "shape": [
197
+ 192
198
+ ],
199
+ "storage": "direct",
200
+ "target": "enc_p.encoder.attn_layers.1.conv_k.bias",
201
+ "transform": "identity"
202
+ },
203
+ {
204
+ "initializer": "enc_p.encoder.attn_layers.1.conv_v.weight",
205
+ "shape": [
206
+ 192,
207
+ 192,
208
+ 1
209
+ ],
210
+ "storage": "direct",
211
+ "target": "enc_p.encoder.attn_layers.1.conv_v.weight",
212
+ "transform": "identity"
213
+ },
214
+ {
215
+ "initializer": "enc_p.encoder.attn_layers.1.conv_v.bias",
216
+ "shape": [
217
+ 192
218
+ ],
219
+ "storage": "direct",
220
+ "target": "enc_p.encoder.attn_layers.1.conv_v.bias",
221
+ "transform": "identity"
222
+ },
223
+ {
224
+ "initializer": "enc_p.encoder.attn_layers.1.conv_o.weight",
225
+ "shape": [
226
+ 192,
227
+ 192,
228
+ 1
229
+ ],
230
+ "storage": "direct",
231
+ "target": "enc_p.encoder.attn_layers.1.conv_o.weight",
232
+ "transform": "identity"
233
+ },
234
+ {
235
+ "initializer": "enc_p.encoder.attn_layers.1.conv_o.bias",
236
+ "shape": [
237
+ 192
238
+ ],
239
+ "storage": "direct",
240
+ "target": "enc_p.encoder.attn_layers.1.conv_o.bias",
241
+ "transform": "identity"
242
+ },
243
+ {
244
+ "initializer": "enc_p.encoder.attn_layers.2.emb_rel_k",
245
+ "shape": [
246
+ 1,
247
+ 9,
248
+ 96
249
+ ],
250
+ "storage": "direct",
251
+ "target": "enc_p.encoder.attn_layers.2.emb_rel_k",
252
+ "transform": "identity"
253
+ },
254
+ {
255
+ "initializer": "enc_p.encoder.attn_layers.2.emb_rel_v",
256
+ "shape": [
257
+ 1,
258
+ 9,
259
+ 96
260
+ ],
261
+ "storage": "direct",
262
+ "target": "enc_p.encoder.attn_layers.2.emb_rel_v",
263
+ "transform": "identity"
264
+ },
265
+ {
266
+ "initializer": "enc_p.encoder.attn_layers.2.conv_q.weight",
267
+ "shape": [
268
+ 192,
269
+ 192,
270
+ 1
271
+ ],
272
+ "storage": "direct",
273
+ "target": "enc_p.encoder.attn_layers.2.conv_q.weight",
274
+ "transform": "identity"
275
+ },
276
+ {
277
+ "initializer": "enc_p.encoder.attn_layers.2.conv_q.bias",
278
+ "shape": [
279
+ 192
280
+ ],
281
+ "storage": "direct",
282
+ "target": "enc_p.encoder.attn_layers.2.conv_q.bias",
283
+ "transform": "identity"
284
+ },
285
+ {
286
+ "initializer": "enc_p.encoder.attn_layers.2.conv_k.weight",
287
+ "shape": [
288
+ 192,
289
+ 192,
290
+ 1
291
+ ],
292
+ "storage": "direct",
293
+ "target": "enc_p.encoder.attn_layers.2.conv_k.weight",
294
+ "transform": "identity"
295
+ },
296
+ {
297
+ "initializer": "enc_p.encoder.attn_layers.2.conv_k.bias",
298
+ "shape": [
299
+ 192
300
+ ],
301
+ "storage": "direct",
302
+ "target": "enc_p.encoder.attn_layers.2.conv_k.bias",
303
+ "transform": "identity"
304
+ },
305
+ {
306
+ "initializer": "enc_p.encoder.attn_layers.2.conv_v.weight",
307
+ "shape": [
308
+ 192,
309
+ 192,
310
+ 1
311
+ ],
312
+ "storage": "direct",
313
+ "target": "enc_p.encoder.attn_layers.2.conv_v.weight",
314
+ "transform": "identity"
315
+ },
316
+ {
317
+ "initializer": "enc_p.encoder.attn_layers.2.conv_v.bias",
318
+ "shape": [
319
+ 192
320
+ ],
321
+ "storage": "direct",
322
+ "target": "enc_p.encoder.attn_layers.2.conv_v.bias",
323
+ "transform": "identity"
324
+ },
325
+ {
326
+ "initializer": "enc_p.encoder.attn_layers.2.conv_o.weight",
327
+ "shape": [
328
+ 192,
329
+ 192,
330
+ 1
331
+ ],
332
+ "storage": "direct",
333
+ "target": "enc_p.encoder.attn_layers.2.conv_o.weight",
334
+ "transform": "identity"
335
+ },
336
+ {
337
+ "initializer": "enc_p.encoder.attn_layers.2.conv_o.bias",
338
+ "shape": [
339
+ 192
340
+ ],
341
+ "storage": "direct",
342
+ "target": "enc_p.encoder.attn_layers.2.conv_o.bias",
343
+ "transform": "identity"
344
+ },
345
+ {
346
+ "initializer": "enc_p.encoder.attn_layers.3.emb_rel_k",
347
+ "shape": [
348
+ 1,
349
+ 9,
350
+ 96
351
+ ],
352
+ "storage": "direct",
353
+ "target": "enc_p.encoder.attn_layers.3.emb_rel_k",
354
+ "transform": "identity"
355
+ },
356
+ {
357
+ "initializer": "enc_p.encoder.attn_layers.3.emb_rel_v",
358
+ "shape": [
359
+ 1,
360
+ 9,
361
+ 96
362
+ ],
363
+ "storage": "direct",
364
+ "target": "enc_p.encoder.attn_layers.3.emb_rel_v",
365
+ "transform": "identity"
366
+ },
367
+ {
368
+ "initializer": "enc_p.encoder.attn_layers.3.conv_q.weight",
369
+ "shape": [
370
+ 192,
371
+ 192,
372
+ 1
373
+ ],
374
+ "storage": "direct",
375
+ "target": "enc_p.encoder.attn_layers.3.conv_q.weight",
376
+ "transform": "identity"
377
+ },
378
+ {
379
+ "initializer": "enc_p.encoder.attn_layers.3.conv_q.bias",
380
+ "shape": [
381
+ 192
382
+ ],
383
+ "storage": "direct",
384
+ "target": "enc_p.encoder.attn_layers.3.conv_q.bias",
385
+ "transform": "identity"
386
+ },
387
+ {
388
+ "initializer": "enc_p.encoder.attn_layers.3.conv_k.weight",
389
+ "shape": [
390
+ 192,
391
+ 192,
392
+ 1
393
+ ],
394
+ "storage": "direct",
395
+ "target": "enc_p.encoder.attn_layers.3.conv_k.weight",
396
+ "transform": "identity"
397
+ },
398
+ {
399
+ "initializer": "enc_p.encoder.attn_layers.3.conv_k.bias",
400
+ "shape": [
401
+ 192
402
+ ],
403
+ "storage": "direct",
404
+ "target": "enc_p.encoder.attn_layers.3.conv_k.bias",
405
+ "transform": "identity"
406
+ },
407
+ {
408
+ "initializer": "enc_p.encoder.attn_layers.3.conv_v.weight",
409
+ "shape": [
410
+ 192,
411
+ 192,
412
+ 1
413
+ ],
414
+ "storage": "direct",
415
+ "target": "enc_p.encoder.attn_layers.3.conv_v.weight",
416
+ "transform": "identity"
417
+ },
418
+ {
419
+ "initializer": "enc_p.encoder.attn_layers.3.conv_v.bias",
420
+ "shape": [
421
+ 192
422
+ ],
423
+ "storage": "direct",
424
+ "target": "enc_p.encoder.attn_layers.3.conv_v.bias",
425
+ "transform": "identity"
426
+ },
427
+ {
428
+ "initializer": "enc_p.encoder.attn_layers.3.conv_o.weight",
429
+ "shape": [
430
+ 192,
431
+ 192,
432
+ 1
433
+ ],
434
+ "storage": "direct",
435
+ "target": "enc_p.encoder.attn_layers.3.conv_o.weight",
436
+ "transform": "identity"
437
+ },
438
+ {
439
+ "initializer": "enc_p.encoder.attn_layers.3.conv_o.bias",
440
+ "shape": [
441
+ 192
442
+ ],
443
+ "storage": "direct",
444
+ "target": "enc_p.encoder.attn_layers.3.conv_o.bias",
445
+ "transform": "identity"
446
+ },
447
+ {
448
+ "initializer": "enc_p.encoder.attn_layers.4.emb_rel_k",
449
+ "shape": [
450
+ 1,
451
+ 9,
452
+ 96
453
+ ],
454
+ "storage": "direct",
455
+ "target": "enc_p.encoder.attn_layers.4.emb_rel_k",
456
+ "transform": "identity"
457
+ },
458
+ {
459
+ "initializer": "enc_p.encoder.attn_layers.4.emb_rel_v",
460
+ "shape": [
461
+ 1,
462
+ 9,
463
+ 96
464
+ ],
465
+ "storage": "direct",
466
+ "target": "enc_p.encoder.attn_layers.4.emb_rel_v",
467
+ "transform": "identity"
468
+ },
469
+ {
470
+ "initializer": "enc_p.encoder.attn_layers.4.conv_q.weight",
471
+ "shape": [
472
+ 192,
473
+ 192,
474
+ 1
475
+ ],
476
+ "storage": "direct",
477
+ "target": "enc_p.encoder.attn_layers.4.conv_q.weight",
478
+ "transform": "identity"
479
+ },
480
+ {
481
+ "initializer": "enc_p.encoder.attn_layers.4.conv_q.bias",
482
+ "shape": [
483
+ 192
484
+ ],
485
+ "storage": "direct",
486
+ "target": "enc_p.encoder.attn_layers.4.conv_q.bias",
487
+ "transform": "identity"
488
+ },
489
+ {
490
+ "initializer": "enc_p.encoder.attn_layers.4.conv_k.weight",
491
+ "shape": [
492
+ 192,
493
+ 192,
494
+ 1
495
+ ],
496
+ "storage": "direct",
497
+ "target": "enc_p.encoder.attn_layers.4.conv_k.weight",
498
+ "transform": "identity"
499
+ },
500
+ {
501
+ "initializer": "enc_p.encoder.attn_layers.4.conv_k.bias",
502
+ "shape": [
503
+ 192
504
+ ],
505
+ "storage": "direct",
506
+ "target": "enc_p.encoder.attn_layers.4.conv_k.bias",
507
+ "transform": "identity"
508
+ },
509
+ {
510
+ "initializer": "enc_p.encoder.attn_layers.4.conv_v.weight",
511
+ "shape": [
512
+ 192,
513
+ 192,
514
+ 1
515
+ ],
516
+ "storage": "direct",
517
+ "target": "enc_p.encoder.attn_layers.4.conv_v.weight",
518
+ "transform": "identity"
519
+ },
520
+ {
521
+ "initializer": "enc_p.encoder.attn_layers.4.conv_v.bias",
522
+ "shape": [
523
+ 192
524
+ ],
525
+ "storage": "direct",
526
+ "target": "enc_p.encoder.attn_layers.4.conv_v.bias",
527
+ "transform": "identity"
528
+ },
529
+ {
530
+ "initializer": "enc_p.encoder.attn_layers.4.conv_o.weight",
531
+ "shape": [
532
+ 192,
533
+ 192,
534
+ 1
535
+ ],
536
+ "storage": "direct",
537
+ "target": "enc_p.encoder.attn_layers.4.conv_o.weight",
538
+ "transform": "identity"
539
+ },
540
+ {
541
+ "initializer": "enc_p.encoder.attn_layers.4.conv_o.bias",
542
+ "shape": [
543
+ 192
544
+ ],
545
+ "storage": "direct",
546
+ "target": "enc_p.encoder.attn_layers.4.conv_o.bias",
547
+ "transform": "identity"
548
+ },
549
+ {
550
+ "initializer": "enc_p.encoder.attn_layers.5.emb_rel_k",
551
+ "shape": [
552
+ 1,
553
+ 9,
554
+ 96
555
+ ],
556
+ "storage": "direct",
557
+ "target": "enc_p.encoder.attn_layers.5.emb_rel_k",
558
+ "transform": "identity"
559
+ },
560
+ {
561
+ "initializer": "enc_p.encoder.attn_layers.5.emb_rel_v",
562
+ "shape": [
563
+ 1,
564
+ 9,
565
+ 96
566
+ ],
567
+ "storage": "direct",
568
+ "target": "enc_p.encoder.attn_layers.5.emb_rel_v",
569
+ "transform": "identity"
570
+ },
571
+ {
572
+ "initializer": "enc_p.encoder.attn_layers.5.conv_q.weight",
573
+ "shape": [
574
+ 192,
575
+ 192,
576
+ 1
577
+ ],
578
+ "storage": "direct",
579
+ "target": "enc_p.encoder.attn_layers.5.conv_q.weight",
580
+ "transform": "identity"
581
+ },
582
+ {
583
+ "initializer": "enc_p.encoder.attn_layers.5.conv_q.bias",
584
+ "shape": [
585
+ 192
586
+ ],
587
+ "storage": "direct",
588
+ "target": "enc_p.encoder.attn_layers.5.conv_q.bias",
589
+ "transform": "identity"
590
+ },
591
+ {
592
+ "initializer": "enc_p.encoder.attn_layers.5.conv_k.weight",
593
+ "shape": [
594
+ 192,
595
+ 192,
596
+ 1
597
+ ],
598
+ "storage": "direct",
599
+ "target": "enc_p.encoder.attn_layers.5.conv_k.weight",
600
+ "transform": "identity"
601
+ },
602
+ {
603
+ "initializer": "enc_p.encoder.attn_layers.5.conv_k.bias",
604
+ "shape": [
605
+ 192
606
+ ],
607
+ "storage": "direct",
608
+ "target": "enc_p.encoder.attn_layers.5.conv_k.bias",
609
+ "transform": "identity"
610
+ },
611
+ {
612
+ "initializer": "enc_p.encoder.attn_layers.5.conv_v.weight",
613
+ "shape": [
614
+ 192,
615
+ 192,
616
+ 1
617
+ ],
618
+ "storage": "direct",
619
+ "target": "enc_p.encoder.attn_layers.5.conv_v.weight",
620
+ "transform": "identity"
621
+ },
622
+ {
623
+ "initializer": "enc_p.encoder.attn_layers.5.conv_v.bias",
624
+ "shape": [
625
+ 192
626
+ ],
627
+ "storage": "direct",
628
+ "target": "enc_p.encoder.attn_layers.5.conv_v.bias",
629
+ "transform": "identity"
630
+ },
631
+ {
632
+ "initializer": "enc_p.encoder.attn_layers.5.conv_o.weight",
633
+ "shape": [
634
+ 192,
635
+ 192,
636
+ 1
637
+ ],
638
+ "storage": "direct",
639
+ "target": "enc_p.encoder.attn_layers.5.conv_o.weight",
640
+ "transform": "identity"
641
+ },
642
+ {
643
+ "initializer": "enc_p.encoder.attn_layers.5.conv_o.bias",
644
+ "shape": [
645
+ 192
646
+ ],
647
+ "storage": "direct",
648
+ "target": "enc_p.encoder.attn_layers.5.conv_o.bias",
649
+ "transform": "identity"
650
+ },
651
+ {
652
+ "initializer": "enc_p.encoder.norm_layers_1.0.gamma",
653
+ "shape": [
654
+ 192
655
+ ],
656
+ "storage": "direct",
657
+ "target": "enc_p.encoder.norm_layers_1.0.gamma",
658
+ "transform": "identity"
659
+ },
660
+ {
661
+ "initializer": "enc_p.encoder.norm_layers_1.0.beta",
662
+ "shape": [
663
+ 192
664
+ ],
665
+ "storage": "direct",
666
+ "target": "enc_p.encoder.norm_layers_1.0.beta",
667
+ "transform": "identity"
668
+ },
669
+ {
670
+ "initializer": "enc_p.encoder.norm_layers_1.1.gamma",
671
+ "shape": [
672
+ 192
673
+ ],
674
+ "storage": "direct",
675
+ "target": "enc_p.encoder.norm_layers_1.1.gamma",
676
+ "transform": "identity"
677
+ },
678
+ {
679
+ "initializer": "enc_p.encoder.norm_layers_1.1.beta",
680
+ "shape": [
681
+ 192
682
+ ],
683
+ "storage": "direct",
684
+ "target": "enc_p.encoder.norm_layers_1.1.beta",
685
+ "transform": "identity"
686
+ },
687
+ {
688
+ "initializer": "enc_p.encoder.norm_layers_1.2.gamma",
689
+ "shape": [
690
+ 192
691
+ ],
692
+ "storage": "direct",
693
+ "target": "enc_p.encoder.norm_layers_1.2.gamma",
694
+ "transform": "identity"
695
+ },
696
+ {
697
+ "initializer": "enc_p.encoder.norm_layers_1.2.beta",
698
+ "shape": [
699
+ 192
700
+ ],
701
+ "storage": "direct",
702
+ "target": "enc_p.encoder.norm_layers_1.2.beta",
703
+ "transform": "identity"
704
+ },
705
+ {
706
+ "initializer": "enc_p.encoder.norm_layers_1.3.gamma",
707
+ "shape": [
708
+ 192
709
+ ],
710
+ "storage": "direct",
711
+ "target": "enc_p.encoder.norm_layers_1.3.gamma",
712
+ "transform": "identity"
713
+ },
714
+ {
715
+ "initializer": "enc_p.encoder.norm_layers_1.3.beta",
716
+ "shape": [
717
+ 192
718
+ ],
719
+ "storage": "direct",
720
+ "target": "enc_p.encoder.norm_layers_1.3.beta",
721
+ "transform": "identity"
722
+ },
723
+ {
724
+ "initializer": "enc_p.encoder.norm_layers_1.4.gamma",
725
+ "shape": [
726
+ 192
727
+ ],
728
+ "storage": "direct",
729
+ "target": "enc_p.encoder.norm_layers_1.4.gamma",
730
+ "transform": "identity"
731
+ },
732
+ {
733
+ "initializer": "enc_p.encoder.norm_layers_1.4.beta",
734
+ "shape": [
735
+ 192
736
+ ],
737
+ "storage": "direct",
738
+ "target": "enc_p.encoder.norm_layers_1.4.beta",
739
+ "transform": "identity"
740
+ },
741
+ {
742
+ "initializer": "enc_p.encoder.norm_layers_1.5.gamma",
743
+ "shape": [
744
+ 192
745
+ ],
746
+ "storage": "direct",
747
+ "target": "enc_p.encoder.norm_layers_1.5.gamma",
748
+ "transform": "identity"
749
+ },
750
+ {
751
+ "initializer": "enc_p.encoder.norm_layers_1.5.beta",
752
+ "shape": [
753
+ 192
754
+ ],
755
+ "storage": "direct",
756
+ "target": "enc_p.encoder.norm_layers_1.5.beta",
757
+ "transform": "identity"
758
+ },
759
+ {
760
+ "initializer": "enc_p.encoder.ffn_layers.0.conv_1.weight",
761
+ "shape": [
762
+ 768,
763
+ 192,
764
+ 3
765
+ ],
766
+ "storage": "direct",
767
+ "target": "enc_p.encoder.ffn_layers.0.conv_1.weight",
768
+ "transform": "identity"
769
+ },
770
+ {
771
+ "initializer": "enc_p.encoder.ffn_layers.0.conv_1.bias",
772
+ "shape": [
773
+ 768
774
+ ],
775
+ "storage": "direct",
776
+ "target": "enc_p.encoder.ffn_layers.0.conv_1.bias",
777
+ "transform": "identity"
778
+ },
779
+ {
780
+ "initializer": "enc_p.encoder.ffn_layers.0.conv_2.weight",
781
+ "shape": [
782
+ 192,
783
+ 768,
784
+ 3
785
+ ],
786
+ "storage": "direct",
787
+ "target": "enc_p.encoder.ffn_layers.0.conv_2.weight",
788
+ "transform": "identity"
789
+ },
790
+ {
791
+ "initializer": "enc_p.encoder.ffn_layers.0.conv_2.bias",
792
+ "shape": [
793
+ 192
794
+ ],
795
+ "storage": "direct",
796
+ "target": "enc_p.encoder.ffn_layers.0.conv_2.bias",
797
+ "transform": "identity"
798
+ },
799
+ {
800
+ "initializer": "enc_p.encoder.ffn_layers.1.conv_1.weight",
801
+ "shape": [
802
+ 768,
803
+ 192,
804
+ 3
805
+ ],
806
+ "storage": "direct",
807
+ "target": "enc_p.encoder.ffn_layers.1.conv_1.weight",
808
+ "transform": "identity"
809
+ },
810
+ {
811
+ "initializer": "enc_p.encoder.ffn_layers.1.conv_1.bias",
812
+ "shape": [
813
+ 768
814
+ ],
815
+ "storage": "direct",
816
+ "target": "enc_p.encoder.ffn_layers.1.conv_1.bias",
817
+ "transform": "identity"
818
+ },
819
+ {
820
+ "initializer": "enc_p.encoder.ffn_layers.1.conv_2.weight",
821
+ "shape": [
822
+ 192,
823
+ 768,
824
+ 3
825
+ ],
826
+ "storage": "direct",
827
+ "target": "enc_p.encoder.ffn_layers.1.conv_2.weight",
828
+ "transform": "identity"
829
+ },
830
+ {
831
+ "initializer": "enc_p.encoder.ffn_layers.1.conv_2.bias",
832
+ "shape": [
833
+ 192
834
+ ],
835
+ "storage": "direct",
836
+ "target": "enc_p.encoder.ffn_layers.1.conv_2.bias",
837
+ "transform": "identity"
838
+ },
839
+ {
840
+ "initializer": "enc_p.encoder.ffn_layers.2.conv_1.weight",
841
+ "shape": [
842
+ 768,
843
+ 192,
844
+ 3
845
+ ],
846
+ "storage": "direct",
847
+ "target": "enc_p.encoder.ffn_layers.2.conv_1.weight",
848
+ "transform": "identity"
849
+ },
850
+ {
851
+ "initializer": "enc_p.encoder.ffn_layers.2.conv_1.bias",
852
+ "shape": [
853
+ 768
854
+ ],
855
+ "storage": "direct",
856
+ "target": "enc_p.encoder.ffn_layers.2.conv_1.bias",
857
+ "transform": "identity"
858
+ },
859
+ {
860
+ "initializer": "enc_p.encoder.ffn_layers.2.conv_2.weight",
861
+ "shape": [
862
+ 192,
863
+ 768,
864
+ 3
865
+ ],
866
+ "storage": "direct",
867
+ "target": "enc_p.encoder.ffn_layers.2.conv_2.weight",
868
+ "transform": "identity"
869
+ },
870
+ {
871
+ "initializer": "enc_p.encoder.ffn_layers.2.conv_2.bias",
872
+ "shape": [
873
+ 192
874
+ ],
875
+ "storage": "direct",
876
+ "target": "enc_p.encoder.ffn_layers.2.conv_2.bias",
877
+ "transform": "identity"
878
+ },
879
+ {
880
+ "initializer": "enc_p.encoder.ffn_layers.3.conv_1.weight",
881
+ "shape": [
882
+ 768,
883
+ 192,
884
+ 3
885
+ ],
886
+ "storage": "direct",
887
+ "target": "enc_p.encoder.ffn_layers.3.conv_1.weight",
888
+ "transform": "identity"
889
+ },
890
+ {
891
+ "initializer": "enc_p.encoder.ffn_layers.3.conv_1.bias",
892
+ "shape": [
893
+ 768
894
+ ],
895
+ "storage": "direct",
896
+ "target": "enc_p.encoder.ffn_layers.3.conv_1.bias",
897
+ "transform": "identity"
898
+ },
899
+ {
900
+ "initializer": "enc_p.encoder.ffn_layers.3.conv_2.weight",
901
+ "shape": [
902
+ 192,
903
+ 768,
904
+ 3
905
+ ],
906
+ "storage": "direct",
907
+ "target": "enc_p.encoder.ffn_layers.3.conv_2.weight",
908
+ "transform": "identity"
909
+ },
910
+ {
911
+ "initializer": "enc_p.encoder.ffn_layers.3.conv_2.bias",
912
+ "shape": [
913
+ 192
914
+ ],
915
+ "storage": "direct",
916
+ "target": "enc_p.encoder.ffn_layers.3.conv_2.bias",
917
+ "transform": "identity"
918
+ },
919
+ {
920
+ "initializer": "enc_p.encoder.ffn_layers.4.conv_1.weight",
921
+ "shape": [
922
+ 768,
923
+ 192,
924
+ 3
925
+ ],
926
+ "storage": "direct",
927
+ "target": "enc_p.encoder.ffn_layers.4.conv_1.weight",
928
+ "transform": "identity"
929
+ },
930
+ {
931
+ "initializer": "enc_p.encoder.ffn_layers.4.conv_1.bias",
932
+ "shape": [
933
+ 768
934
+ ],
935
+ "storage": "direct",
936
+ "target": "enc_p.encoder.ffn_layers.4.conv_1.bias",
937
+ "transform": "identity"
938
+ },
939
+ {
940
+ "initializer": "enc_p.encoder.ffn_layers.4.conv_2.weight",
941
+ "shape": [
942
+ 192,
943
+ 768,
944
+ 3
945
+ ],
946
+ "storage": "direct",
947
+ "target": "enc_p.encoder.ffn_layers.4.conv_2.weight",
948
+ "transform": "identity"
949
+ },
950
+ {
951
+ "initializer": "enc_p.encoder.ffn_layers.4.conv_2.bias",
952
+ "shape": [
953
+ 192
954
+ ],
955
+ "storage": "direct",
956
+ "target": "enc_p.encoder.ffn_layers.4.conv_2.bias",
957
+ "transform": "identity"
958
+ },
959
+ {
960
+ "initializer": "enc_p.encoder.ffn_layers.5.conv_1.weight",
961
+ "shape": [
962
+ 768,
963
+ 192,
964
+ 3
965
+ ],
966
+ "storage": "direct",
967
+ "target": "enc_p.encoder.ffn_layers.5.conv_1.weight",
968
+ "transform": "identity"
969
+ },
970
+ {
971
+ "initializer": "enc_p.encoder.ffn_layers.5.conv_1.bias",
972
+ "shape": [
973
+ 768
974
+ ],
975
+ "storage": "direct",
976
+ "target": "enc_p.encoder.ffn_layers.5.conv_1.bias",
977
+ "transform": "identity"
978
+ },
979
+ {
980
+ "initializer": "enc_p.encoder.ffn_layers.5.conv_2.weight",
981
+ "shape": [
982
+ 192,
983
+ 768,
984
+ 3
985
+ ],
986
+ "storage": "direct",
987
+ "target": "enc_p.encoder.ffn_layers.5.conv_2.weight",
988
+ "transform": "identity"
989
+ },
990
+ {
991
+ "initializer": "enc_p.encoder.ffn_layers.5.conv_2.bias",
992
+ "shape": [
993
+ 192
994
+ ],
995
+ "storage": "direct",
996
+ "target": "enc_p.encoder.ffn_layers.5.conv_2.bias",
997
+ "transform": "identity"
998
+ },
999
+ {
1000
+ "initializer": "enc_p.encoder.norm_layers_2.0.gamma",
1001
+ "shape": [
1002
+ 192
1003
+ ],
1004
+ "storage": "direct",
1005
+ "target": "enc_p.encoder.norm_layers_2.0.gamma",
1006
+ "transform": "identity"
1007
+ },
1008
+ {
1009
+ "initializer": "enc_p.encoder.norm_layers_2.0.beta",
1010
+ "shape": [
1011
+ 192
1012
+ ],
1013
+ "storage": "direct",
1014
+ "target": "enc_p.encoder.norm_layers_2.0.beta",
1015
+ "transform": "identity"
1016
+ },
1017
+ {
1018
+ "initializer": "enc_p.encoder.norm_layers_2.1.gamma",
1019
+ "shape": [
1020
+ 192
1021
+ ],
1022
+ "storage": "direct",
1023
+ "target": "enc_p.encoder.norm_layers_2.1.gamma",
1024
+ "transform": "identity"
1025
+ },
1026
+ {
1027
+ "initializer": "enc_p.encoder.norm_layers_2.1.beta",
1028
+ "shape": [
1029
+ 192
1030
+ ],
1031
+ "storage": "direct",
1032
+ "target": "enc_p.encoder.norm_layers_2.1.beta",
1033
+ "transform": "identity"
1034
+ },
1035
+ {
1036
+ "initializer": "enc_p.encoder.norm_layers_2.2.gamma",
1037
+ "shape": [
1038
+ 192
1039
+ ],
1040
+ "storage": "direct",
1041
+ "target": "enc_p.encoder.norm_layers_2.2.gamma",
1042
+ "transform": "identity"
1043
+ },
1044
+ {
1045
+ "initializer": "enc_p.encoder.norm_layers_2.2.beta",
1046
+ "shape": [
1047
+ 192
1048
+ ],
1049
+ "storage": "direct",
1050
+ "target": "enc_p.encoder.norm_layers_2.2.beta",
1051
+ "transform": "identity"
1052
+ },
1053
+ {
1054
+ "initializer": "enc_p.encoder.norm_layers_2.3.gamma",
1055
+ "shape": [
1056
+ 192
1057
+ ],
1058
+ "storage": "direct",
1059
+ "target": "enc_p.encoder.norm_layers_2.3.gamma",
1060
+ "transform": "identity"
1061
+ },
1062
+ {
1063
+ "initializer": "enc_p.encoder.norm_layers_2.3.beta",
1064
+ "shape": [
1065
+ 192
1066
+ ],
1067
+ "storage": "direct",
1068
+ "target": "enc_p.encoder.norm_layers_2.3.beta",
1069
+ "transform": "identity"
1070
+ },
1071
+ {
1072
+ "initializer": "enc_p.encoder.norm_layers_2.4.gamma",
1073
+ "shape": [
1074
+ 192
1075
+ ],
1076
+ "storage": "direct",
1077
+ "target": "enc_p.encoder.norm_layers_2.4.gamma",
1078
+ "transform": "identity"
1079
+ },
1080
+ {
1081
+ "initializer": "enc_p.encoder.norm_layers_2.4.beta",
1082
+ "shape": [
1083
+ 192
1084
+ ],
1085
+ "storage": "direct",
1086
+ "target": "enc_p.encoder.norm_layers_2.4.beta",
1087
+ "transform": "identity"
1088
+ },
1089
+ {
1090
+ "initializer": "enc_p.encoder.norm_layers_2.5.gamma",
1091
+ "shape": [
1092
+ 192
1093
+ ],
1094
+ "storage": "direct",
1095
+ "target": "enc_p.encoder.norm_layers_2.5.gamma",
1096
+ "transform": "identity"
1097
+ },
1098
+ {
1099
+ "initializer": "enc_p.encoder.norm_layers_2.5.beta",
1100
+ "shape": [
1101
+ 192
1102
+ ],
1103
+ "storage": "direct",
1104
+ "target": "enc_p.encoder.norm_layers_2.5.beta",
1105
+ "transform": "identity"
1106
+ },
1107
+ {
1108
+ "initializer": "enc_p.proj.weight",
1109
+ "shape": [
1110
+ 384,
1111
+ 192,
1112
+ 1
1113
+ ],
1114
+ "storage": "direct",
1115
+ "target": "enc_p.proj.weight",
1116
+ "transform": "identity"
1117
+ },
1118
+ {
1119
+ "initializer": "enc_p.proj.bias",
1120
+ "shape": [
1121
+ 384
1122
+ ],
1123
+ "storage": "direct",
1124
+ "target": "enc_p.proj.bias",
1125
+ "transform": "identity"
1126
+ },
1127
+ {
1128
+ "initializer": "dec.conv_pre.weight",
1129
+ "shape": [
1130
+ 256,
1131
+ 192,
1132
+ 7
1133
+ ],
1134
+ "storage": "direct",
1135
+ "target": "dec.conv_pre.weight",
1136
+ "transform": "identity"
1137
+ },
1138
+ {
1139
+ "initializer": "dec.conv_pre.bias",
1140
+ "shape": [
1141
+ 256
1142
+ ],
1143
+ "storage": "direct",
1144
+ "target": "dec.conv_pre.bias",
1145
+ "transform": "identity"
1146
+ },
1147
+ {
1148
+ "initializer": "dec.ups.0.bias",
1149
+ "shape": [
1150
+ 128
1151
+ ],
1152
+ "storage": "direct",
1153
+ "target": "dec.ups.0.bias",
1154
+ "transform": "identity"
1155
+ },
1156
+ {
1157
+ "initializer": "dec.ups.0.weight",
1158
+ "shape": [
1159
+ 256,
1160
+ 128,
1161
+ 16
1162
+ ],
1163
+ "storage": "weight_norm_reconstructed",
1164
+ "target": "dec.ups.0.weight",
1165
+ "transform": "identity"
1166
+ },
1167
+ {
1168
+ "initializer": "dec.ups.1.bias",
1169
+ "shape": [
1170
+ 64
1171
+ ],
1172
+ "storage": "direct",
1173
+ "target": "dec.ups.1.bias",
1174
+ "transform": "identity"
1175
+ },
1176
+ {
1177
+ "initializer": "dec.ups.1.weight",
1178
+ "shape": [
1179
+ 128,
1180
+ 64,
1181
+ 16
1182
+ ],
1183
+ "storage": "weight_norm_reconstructed",
1184
+ "target": "dec.ups.1.weight",
1185
+ "transform": "identity"
1186
+ },
1187
+ {
1188
+ "initializer": "dec.ups.2.bias",
1189
+ "shape": [
1190
+ 32
1191
+ ],
1192
+ "storage": "direct",
1193
+ "target": "dec.ups.2.bias",
1194
+ "transform": "identity"
1195
+ },
1196
+ {
1197
+ "initializer": "dec.ups.2.weight",
1198
+ "shape": [
1199
+ 64,
1200
+ 32,
1201
+ 8
1202
+ ],
1203
+ "storage": "weight_norm_reconstructed",
1204
+ "target": "dec.ups.2.weight",
1205
+ "transform": "identity"
1206
+ },
1207
+ {
1208
+ "initializer": "dec.resblocks.0.convs.0.bias",
1209
+ "shape": [
1210
+ 128
1211
+ ],
1212
+ "storage": "direct",
1213
+ "target": "dec.resblocks.0.convs.0.bias",
1214
+ "transform": "identity"
1215
+ },
1216
+ {
1217
+ "initializer": "dec.resblocks.0.convs.0.weight",
1218
+ "shape": [
1219
+ 128,
1220
+ 128,
1221
+ 3
1222
+ ],
1223
+ "storage": "weight_norm_reconstructed",
1224
+ "target": "dec.resblocks.0.convs.0.weight",
1225
+ "transform": "identity"
1226
+ },
1227
+ {
1228
+ "initializer": "dec.resblocks.0.convs.1.bias",
1229
+ "shape": [
1230
+ 128
1231
+ ],
1232
+ "storage": "direct",
1233
+ "target": "dec.resblocks.0.convs.1.bias",
1234
+ "transform": "identity"
1235
+ },
1236
+ {
1237
+ "initializer": "dec.resblocks.0.convs.1.weight",
1238
+ "shape": [
1239
+ 128,
1240
+ 128,
1241
+ 3
1242
+ ],
1243
+ "storage": "weight_norm_reconstructed",
1244
+ "target": "dec.resblocks.0.convs.1.weight",
1245
+ "transform": "identity"
1246
+ },
1247
+ {
1248
+ "initializer": "dec.resblocks.1.convs.0.bias",
1249
+ "shape": [
1250
+ 128
1251
+ ],
1252
+ "storage": "direct",
1253
+ "target": "dec.resblocks.1.convs.0.bias",
1254
+ "transform": "identity"
1255
+ },
1256
+ {
1257
+ "initializer": "dec.resblocks.1.convs.0.weight",
1258
+ "shape": [
1259
+ 128,
1260
+ 128,
1261
+ 5
1262
+ ],
1263
+ "storage": "weight_norm_reconstructed",
1264
+ "target": "dec.resblocks.1.convs.0.weight",
1265
+ "transform": "identity"
1266
+ },
1267
+ {
1268
+ "initializer": "dec.resblocks.1.convs.1.bias",
1269
+ "shape": [
1270
+ 128
1271
+ ],
1272
+ "storage": "direct",
1273
+ "target": "dec.resblocks.1.convs.1.bias",
1274
+ "transform": "identity"
1275
+ },
1276
+ {
1277
+ "initializer": "dec.resblocks.1.convs.1.weight",
1278
+ "shape": [
1279
+ 128,
1280
+ 128,
1281
+ 5
1282
+ ],
1283
+ "storage": "weight_norm_reconstructed",
1284
+ "target": "dec.resblocks.1.convs.1.weight",
1285
+ "transform": "identity"
1286
+ },
1287
+ {
1288
+ "initializer": "dec.resblocks.2.convs.0.bias",
1289
+ "shape": [
1290
+ 128
1291
+ ],
1292
+ "storage": "direct",
1293
+ "target": "dec.resblocks.2.convs.0.bias",
1294
+ "transform": "identity"
1295
+ },
1296
+ {
1297
+ "initializer": "dec.resblocks.2.convs.0.weight",
1298
+ "shape": [
1299
+ 128,
1300
+ 128,
1301
+ 7
1302
+ ],
1303
+ "storage": "weight_norm_reconstructed",
1304
+ "target": "dec.resblocks.2.convs.0.weight",
1305
+ "transform": "identity"
1306
+ },
1307
+ {
1308
+ "initializer": "dec.resblocks.2.convs.1.bias",
1309
+ "shape": [
1310
+ 128
1311
+ ],
1312
+ "storage": "direct",
1313
+ "target": "dec.resblocks.2.convs.1.bias",
1314
+ "transform": "identity"
1315
+ },
1316
+ {
1317
+ "initializer": "dec.resblocks.2.convs.1.weight",
1318
+ "shape": [
1319
+ 128,
1320
+ 128,
1321
+ 7
1322
+ ],
1323
+ "storage": "weight_norm_reconstructed",
1324
+ "target": "dec.resblocks.2.convs.1.weight",
1325
+ "transform": "identity"
1326
+ },
1327
+ {
1328
+ "initializer": "dec.resblocks.3.convs.0.bias",
1329
+ "shape": [
1330
+ 64
1331
+ ],
1332
+ "storage": "direct",
1333
+ "target": "dec.resblocks.3.convs.0.bias",
1334
+ "transform": "identity"
1335
+ },
1336
+ {
1337
+ "initializer": "dec.resblocks.3.convs.0.weight",
1338
+ "shape": [
1339
+ 64,
1340
+ 64,
1341
+ 3
1342
+ ],
1343
+ "storage": "weight_norm_reconstructed",
1344
+ "target": "dec.resblocks.3.convs.0.weight",
1345
+ "transform": "identity"
1346
+ },
1347
+ {
1348
+ "initializer": "dec.resblocks.3.convs.1.bias",
1349
+ "shape": [
1350
+ 64
1351
+ ],
1352
+ "storage": "direct",
1353
+ "target": "dec.resblocks.3.convs.1.bias",
1354
+ "transform": "identity"
1355
+ },
1356
+ {
1357
+ "initializer": "dec.resblocks.3.convs.1.weight",
1358
+ "shape": [
1359
+ 64,
1360
+ 64,
1361
+ 3
1362
+ ],
1363
+ "storage": "weight_norm_reconstructed",
1364
+ "target": "dec.resblocks.3.convs.1.weight",
1365
+ "transform": "identity"
1366
+ },
1367
+ {
1368
+ "initializer": "dec.resblocks.4.convs.0.bias",
1369
+ "shape": [
1370
+ 64
1371
+ ],
1372
+ "storage": "direct",
1373
+ "target": "dec.resblocks.4.convs.0.bias",
1374
+ "transform": "identity"
1375
+ },
1376
+ {
1377
+ "initializer": "dec.resblocks.4.convs.0.weight",
1378
+ "shape": [
1379
+ 64,
1380
+ 64,
1381
+ 5
1382
+ ],
1383
+ "storage": "weight_norm_reconstructed",
1384
+ "target": "dec.resblocks.4.convs.0.weight",
1385
+ "transform": "identity"
1386
+ },
1387
+ {
1388
+ "initializer": "dec.resblocks.4.convs.1.bias",
1389
+ "shape": [
1390
+ 64
1391
+ ],
1392
+ "storage": "direct",
1393
+ "target": "dec.resblocks.4.convs.1.bias",
1394
+ "transform": "identity"
1395
+ },
1396
+ {
1397
+ "initializer": "dec.resblocks.4.convs.1.weight",
1398
+ "shape": [
1399
+ 64,
1400
+ 64,
1401
+ 5
1402
+ ],
1403
+ "storage": "weight_norm_reconstructed",
1404
+ "target": "dec.resblocks.4.convs.1.weight",
1405
+ "transform": "identity"
1406
+ },
1407
+ {
1408
+ "initializer": "dec.resblocks.5.convs.0.bias",
1409
+ "shape": [
1410
+ 64
1411
+ ],
1412
+ "storage": "direct",
1413
+ "target": "dec.resblocks.5.convs.0.bias",
1414
+ "transform": "identity"
1415
+ },
1416
+ {
1417
+ "initializer": "dec.resblocks.5.convs.0.weight",
1418
+ "shape": [
1419
+ 64,
1420
+ 64,
1421
+ 7
1422
+ ],
1423
+ "storage": "weight_norm_reconstructed",
1424
+ "target": "dec.resblocks.5.convs.0.weight",
1425
+ "transform": "identity"
1426
+ },
1427
+ {
1428
+ "initializer": "dec.resblocks.5.convs.1.bias",
1429
+ "shape": [
1430
+ 64
1431
+ ],
1432
+ "storage": "direct",
1433
+ "target": "dec.resblocks.5.convs.1.bias",
1434
+ "transform": "identity"
1435
+ },
1436
+ {
1437
+ "initializer": "dec.resblocks.5.convs.1.weight",
1438
+ "shape": [
1439
+ 64,
1440
+ 64,
1441
+ 7
1442
+ ],
1443
+ "storage": "weight_norm_reconstructed",
1444
+ "target": "dec.resblocks.5.convs.1.weight",
1445
+ "transform": "identity"
1446
+ },
1447
+ {
1448
+ "initializer": "dec.resblocks.6.convs.0.bias",
1449
+ "shape": [
1450
+ 32
1451
+ ],
1452
+ "storage": "direct",
1453
+ "target": "dec.resblocks.6.convs.0.bias",
1454
+ "transform": "identity"
1455
+ },
1456
+ {
1457
+ "initializer": "dec.resblocks.6.convs.0.weight",
1458
+ "shape": [
1459
+ 32,
1460
+ 32,
1461
+ 3
1462
+ ],
1463
+ "storage": "weight_norm_reconstructed",
1464
+ "target": "dec.resblocks.6.convs.0.weight",
1465
+ "transform": "identity"
1466
+ },
1467
+ {
1468
+ "initializer": "dec.resblocks.6.convs.1.bias",
1469
+ "shape": [
1470
+ 32
1471
+ ],
1472
+ "storage": "direct",
1473
+ "target": "dec.resblocks.6.convs.1.bias",
1474
+ "transform": "identity"
1475
+ },
1476
+ {
1477
+ "initializer": "dec.resblocks.6.convs.1.weight",
1478
+ "shape": [
1479
+ 32,
1480
+ 32,
1481
+ 3
1482
+ ],
1483
+ "storage": "weight_norm_reconstructed",
1484
+ "target": "dec.resblocks.6.convs.1.weight",
1485
+ "transform": "identity"
1486
+ },
1487
+ {
1488
+ "initializer": "dec.resblocks.7.convs.0.bias",
1489
+ "shape": [
1490
+ 32
1491
+ ],
1492
+ "storage": "direct",
1493
+ "target": "dec.resblocks.7.convs.0.bias",
1494
+ "transform": "identity"
1495
+ },
1496
+ {
1497
+ "initializer": "dec.resblocks.7.convs.0.weight",
1498
+ "shape": [
1499
+ 32,
1500
+ 32,
1501
+ 5
1502
+ ],
1503
+ "storage": "weight_norm_reconstructed",
1504
+ "target": "dec.resblocks.7.convs.0.weight",
1505
+ "transform": "identity"
1506
+ },
1507
+ {
1508
+ "initializer": "dec.resblocks.7.convs.1.bias",
1509
+ "shape": [
1510
+ 32
1511
+ ],
1512
+ "storage": "direct",
1513
+ "target": "dec.resblocks.7.convs.1.bias",
1514
+ "transform": "identity"
1515
+ },
1516
+ {
1517
+ "initializer": "dec.resblocks.7.convs.1.weight",
1518
+ "shape": [
1519
+ 32,
1520
+ 32,
1521
+ 5
1522
+ ],
1523
+ "storage": "weight_norm_reconstructed",
1524
+ "target": "dec.resblocks.7.convs.1.weight",
1525
+ "transform": "identity"
1526
+ },
1527
+ {
1528
+ "initializer": "dec.resblocks.8.convs.0.bias",
1529
+ "shape": [
1530
+ 32
1531
+ ],
1532
+ "storage": "direct",
1533
+ "target": "dec.resblocks.8.convs.0.bias",
1534
+ "transform": "identity"
1535
+ },
1536
+ {
1537
+ "initializer": "dec.resblocks.8.convs.0.weight",
1538
+ "shape": [
1539
+ 32,
1540
+ 32,
1541
+ 7
1542
+ ],
1543
+ "storage": "weight_norm_reconstructed",
1544
+ "target": "dec.resblocks.8.convs.0.weight",
1545
+ "transform": "identity"
1546
+ },
1547
+ {
1548
+ "initializer": "dec.resblocks.8.convs.1.bias",
1549
+ "shape": [
1550
+ 32
1551
+ ],
1552
+ "storage": "direct",
1553
+ "target": "dec.resblocks.8.convs.1.bias",
1554
+ "transform": "identity"
1555
+ },
1556
+ {
1557
+ "initializer": "dec.resblocks.8.convs.1.weight",
1558
+ "shape": [
1559
+ 32,
1560
+ 32,
1561
+ 7
1562
+ ],
1563
+ "storage": "weight_norm_reconstructed",
1564
+ "target": "dec.resblocks.8.convs.1.weight",
1565
+ "transform": "identity"
1566
+ },
1567
+ {
1568
+ "initializer": "dec.conv_post.weight",
1569
+ "shape": [
1570
+ 1,
1571
+ 32,
1572
+ 7
1573
+ ],
1574
+ "storage": "direct",
1575
+ "target": "dec.conv_post.weight",
1576
+ "transform": "identity"
1577
+ },
1578
+ {
1579
+ "initializer": "flow.flows.0.pre.weight",
1580
+ "shape": [
1581
+ 192,
1582
+ 96,
1583
+ 1
1584
+ ],
1585
+ "storage": "direct",
1586
+ "target": "flow.flows.0.pre.weight",
1587
+ "transform": "identity"
1588
+ },
1589
+ {
1590
+ "initializer": "flow.flows.0.pre.bias",
1591
+ "shape": [
1592
+ 192
1593
+ ],
1594
+ "storage": "direct",
1595
+ "target": "flow.flows.0.pre.bias",
1596
+ "transform": "identity"
1597
+ },
1598
+ {
1599
+ "initializer": "flow.flows.0.enc.in_layers.0.bias",
1600
+ "shape": [
1601
+ 384
1602
+ ],
1603
+ "storage": "direct",
1604
+ "target": "flow.flows.0.enc.in_layers.0.bias",
1605
+ "transform": "identity"
1606
+ },
1607
+ {
1608
+ "initializer": "flow.flows.0.enc.in_layers.0.weight",
1609
+ "shape": [
1610
+ 384,
1611
+ 192,
1612
+ 5
1613
+ ],
1614
+ "storage": "weight_norm_reconstructed",
1615
+ "target": "flow.flows.0.enc.in_layers.0.weight",
1616
+ "transform": "identity"
1617
+ },
1618
+ {
1619
+ "initializer": "flow.flows.0.enc.in_layers.1.bias",
1620
+ "shape": [
1621
+ 384
1622
+ ],
1623
+ "storage": "direct",
1624
+ "target": "flow.flows.0.enc.in_layers.1.bias",
1625
+ "transform": "identity"
1626
+ },
1627
+ {
1628
+ "initializer": "flow.flows.0.enc.in_layers.1.weight",
1629
+ "shape": [
1630
+ 384,
1631
+ 192,
1632
+ 5
1633
+ ],
1634
+ "storage": "weight_norm_reconstructed",
1635
+ "target": "flow.flows.0.enc.in_layers.1.weight",
1636
+ "transform": "identity"
1637
+ },
1638
+ {
1639
+ "initializer": "flow.flows.0.enc.in_layers.2.bias",
1640
+ "shape": [
1641
+ 384
1642
+ ],
1643
+ "storage": "direct",
1644
+ "target": "flow.flows.0.enc.in_layers.2.bias",
1645
+ "transform": "identity"
1646
+ },
1647
+ {
1648
+ "initializer": "flow.flows.0.enc.in_layers.2.weight",
1649
+ "shape": [
1650
+ 384,
1651
+ 192,
1652
+ 5
1653
+ ],
1654
+ "storage": "weight_norm_reconstructed",
1655
+ "target": "flow.flows.0.enc.in_layers.2.weight",
1656
+ "transform": "identity"
1657
+ },
1658
+ {
1659
+ "initializer": "flow.flows.0.enc.in_layers.3.bias",
1660
+ "shape": [
1661
+ 384
1662
+ ],
1663
+ "storage": "direct",
1664
+ "target": "flow.flows.0.enc.in_layers.3.bias",
1665
+ "transform": "identity"
1666
+ },
1667
+ {
1668
+ "initializer": "flow.flows.0.enc.in_layers.3.weight",
1669
+ "shape": [
1670
+ 384,
1671
+ 192,
1672
+ 5
1673
+ ],
1674
+ "storage": "weight_norm_reconstructed",
1675
+ "target": "flow.flows.0.enc.in_layers.3.weight",
1676
+ "transform": "identity"
1677
+ },
1678
+ {
1679
+ "initializer": "flow.flows.0.enc.res_skip_layers.0.bias",
1680
+ "shape": [
1681
+ 384
1682
+ ],
1683
+ "storage": "direct",
1684
+ "target": "flow.flows.0.enc.res_skip_layers.0.bias",
1685
+ "transform": "identity"
1686
+ },
1687
+ {
1688
+ "initializer": "flow.flows.0.enc.res_skip_layers.0.weight",
1689
+ "shape": [
1690
+ 384,
1691
+ 192,
1692
+ 1
1693
+ ],
1694
+ "storage": "weight_norm_reconstructed",
1695
+ "target": "flow.flows.0.enc.res_skip_layers.0.weight",
1696
+ "transform": "identity"
1697
+ },
1698
+ {
1699
+ "initializer": "flow.flows.0.enc.res_skip_layers.1.bias",
1700
+ "shape": [
1701
+ 384
1702
+ ],
1703
+ "storage": "direct",
1704
+ "target": "flow.flows.0.enc.res_skip_layers.1.bias",
1705
+ "transform": "identity"
1706
+ },
1707
+ {
1708
+ "initializer": "flow.flows.0.enc.res_skip_layers.1.weight",
1709
+ "shape": [
1710
+ 384,
1711
+ 192,
1712
+ 1
1713
+ ],
1714
+ "storage": "weight_norm_reconstructed",
1715
+ "target": "flow.flows.0.enc.res_skip_layers.1.weight",
1716
+ "transform": "identity"
1717
+ },
1718
+ {
1719
+ "initializer": "flow.flows.0.enc.res_skip_layers.2.bias",
1720
+ "shape": [
1721
+ 384
1722
+ ],
1723
+ "storage": "direct",
1724
+ "target": "flow.flows.0.enc.res_skip_layers.2.bias",
1725
+ "transform": "identity"
1726
+ },
1727
+ {
1728
+ "initializer": "flow.flows.0.enc.res_skip_layers.2.weight",
1729
+ "shape": [
1730
+ 384,
1731
+ 192,
1732
+ 1
1733
+ ],
1734
+ "storage": "weight_norm_reconstructed",
1735
+ "target": "flow.flows.0.enc.res_skip_layers.2.weight",
1736
+ "transform": "identity"
1737
+ },
1738
+ {
1739
+ "initializer": "flow.flows.0.enc.res_skip_layers.3.bias",
1740
+ "shape": [
1741
+ 192
1742
+ ],
1743
+ "storage": "direct",
1744
+ "target": "flow.flows.0.enc.res_skip_layers.3.bias",
1745
+ "transform": "identity"
1746
+ },
1747
+ {
1748
+ "initializer": "flow.flows.0.enc.res_skip_layers.3.weight",
1749
+ "shape": [
1750
+ 192,
1751
+ 192,
1752
+ 1
1753
+ ],
1754
+ "storage": "weight_norm_reconstructed",
1755
+ "target": "flow.flows.0.enc.res_skip_layers.3.weight",
1756
+ "transform": "identity"
1757
+ },
1758
+ {
1759
+ "initializer": "flow.flows.0.post.weight",
1760
+ "shape": [
1761
+ 96,
1762
+ 192,
1763
+ 1
1764
+ ],
1765
+ "storage": "direct",
1766
+ "target": "flow.flows.0.post.weight",
1767
+ "transform": "identity"
1768
+ },
1769
+ {
1770
+ "initializer": "flow.flows.0.post.bias",
1771
+ "shape": [
1772
+ 96
1773
+ ],
1774
+ "storage": "direct",
1775
+ "target": "flow.flows.0.post.bias",
1776
+ "transform": "identity"
1777
+ },
1778
+ {
1779
+ "initializer": "flow.flows.2.pre.weight",
1780
+ "shape": [
1781
+ 192,
1782
+ 96,
1783
+ 1
1784
+ ],
1785
+ "storage": "direct",
1786
+ "target": "flow.flows.2.pre.weight",
1787
+ "transform": "identity"
1788
+ },
1789
+ {
1790
+ "initializer": "flow.flows.2.pre.bias",
1791
+ "shape": [
1792
+ 192
1793
+ ],
1794
+ "storage": "direct",
1795
+ "target": "flow.flows.2.pre.bias",
1796
+ "transform": "identity"
1797
+ },
1798
+ {
1799
+ "initializer": "flow.flows.2.enc.in_layers.0.bias",
1800
+ "shape": [
1801
+ 384
1802
+ ],
1803
+ "storage": "direct",
1804
+ "target": "flow.flows.2.enc.in_layers.0.bias",
1805
+ "transform": "identity"
1806
+ },
1807
+ {
1808
+ "initializer": "flow.flows.2.enc.in_layers.0.weight",
1809
+ "shape": [
1810
+ 384,
1811
+ 192,
1812
+ 5
1813
+ ],
1814
+ "storage": "weight_norm_reconstructed",
1815
+ "target": "flow.flows.2.enc.in_layers.0.weight",
1816
+ "transform": "identity"
1817
+ },
1818
+ {
1819
+ "initializer": "flow.flows.2.enc.in_layers.1.bias",
1820
+ "shape": [
1821
+ 384
1822
+ ],
1823
+ "storage": "direct",
1824
+ "target": "flow.flows.2.enc.in_layers.1.bias",
1825
+ "transform": "identity"
1826
+ },
1827
+ {
1828
+ "initializer": "flow.flows.2.enc.in_layers.1.weight",
1829
+ "shape": [
1830
+ 384,
1831
+ 192,
1832
+ 5
1833
+ ],
1834
+ "storage": "weight_norm_reconstructed",
1835
+ "target": "flow.flows.2.enc.in_layers.1.weight",
1836
+ "transform": "identity"
1837
+ },
1838
+ {
1839
+ "initializer": "flow.flows.2.enc.in_layers.2.bias",
1840
+ "shape": [
1841
+ 384
1842
+ ],
1843
+ "storage": "direct",
1844
+ "target": "flow.flows.2.enc.in_layers.2.bias",
1845
+ "transform": "identity"
1846
+ },
1847
+ {
1848
+ "initializer": "flow.flows.2.enc.in_layers.2.weight",
1849
+ "shape": [
1850
+ 384,
1851
+ 192,
1852
+ 5
1853
+ ],
1854
+ "storage": "weight_norm_reconstructed",
1855
+ "target": "flow.flows.2.enc.in_layers.2.weight",
1856
+ "transform": "identity"
1857
+ },
1858
+ {
1859
+ "initializer": "flow.flows.2.enc.in_layers.3.bias",
1860
+ "shape": [
1861
+ 384
1862
+ ],
1863
+ "storage": "direct",
1864
+ "target": "flow.flows.2.enc.in_layers.3.bias",
1865
+ "transform": "identity"
1866
+ },
1867
+ {
1868
+ "initializer": "flow.flows.2.enc.in_layers.3.weight",
1869
+ "shape": [
1870
+ 384,
1871
+ 192,
1872
+ 5
1873
+ ],
1874
+ "storage": "weight_norm_reconstructed",
1875
+ "target": "flow.flows.2.enc.in_layers.3.weight",
1876
+ "transform": "identity"
1877
+ },
1878
+ {
1879
+ "initializer": "flow.flows.2.enc.res_skip_layers.0.bias",
1880
+ "shape": [
1881
+ 384
1882
+ ],
1883
+ "storage": "direct",
1884
+ "target": "flow.flows.2.enc.res_skip_layers.0.bias",
1885
+ "transform": "identity"
1886
+ },
1887
+ {
1888
+ "initializer": "flow.flows.2.enc.res_skip_layers.0.weight",
1889
+ "shape": [
1890
+ 384,
1891
+ 192,
1892
+ 1
1893
+ ],
1894
+ "storage": "weight_norm_reconstructed",
1895
+ "target": "flow.flows.2.enc.res_skip_layers.0.weight",
1896
+ "transform": "identity"
1897
+ },
1898
+ {
1899
+ "initializer": "flow.flows.2.enc.res_skip_layers.1.bias",
1900
+ "shape": [
1901
+ 384
1902
+ ],
1903
+ "storage": "direct",
1904
+ "target": "flow.flows.2.enc.res_skip_layers.1.bias",
1905
+ "transform": "identity"
1906
+ },
1907
+ {
1908
+ "initializer": "flow.flows.2.enc.res_skip_layers.1.weight",
1909
+ "shape": [
1910
+ 384,
1911
+ 192,
1912
+ 1
1913
+ ],
1914
+ "storage": "weight_norm_reconstructed",
1915
+ "target": "flow.flows.2.enc.res_skip_layers.1.weight",
1916
+ "transform": "identity"
1917
+ },
1918
+ {
1919
+ "initializer": "flow.flows.2.enc.res_skip_layers.2.bias",
1920
+ "shape": [
1921
+ 384
1922
+ ],
1923
+ "storage": "direct",
1924
+ "target": "flow.flows.2.enc.res_skip_layers.2.bias",
1925
+ "transform": "identity"
1926
+ },
1927
+ {
1928
+ "initializer": "flow.flows.2.enc.res_skip_layers.2.weight",
1929
+ "shape": [
1930
+ 384,
1931
+ 192,
1932
+ 1
1933
+ ],
1934
+ "storage": "weight_norm_reconstructed",
1935
+ "target": "flow.flows.2.enc.res_skip_layers.2.weight",
1936
+ "transform": "identity"
1937
+ },
1938
+ {
1939
+ "initializer": "flow.flows.2.enc.res_skip_layers.3.bias",
1940
+ "shape": [
1941
+ 192
1942
+ ],
1943
+ "storage": "direct",
1944
+ "target": "flow.flows.2.enc.res_skip_layers.3.bias",
1945
+ "transform": "identity"
1946
+ },
1947
+ {
1948
+ "initializer": "flow.flows.2.enc.res_skip_layers.3.weight",
1949
+ "shape": [
1950
+ 192,
1951
+ 192,
1952
+ 1
1953
+ ],
1954
+ "storage": "weight_norm_reconstructed",
1955
+ "target": "flow.flows.2.enc.res_skip_layers.3.weight",
1956
+ "transform": "identity"
1957
+ },
1958
+ {
1959
+ "initializer": "flow.flows.2.post.weight",
1960
+ "shape": [
1961
+ 96,
1962
+ 192,
1963
+ 1
1964
+ ],
1965
+ "storage": "direct",
1966
+ "target": "flow.flows.2.post.weight",
1967
+ "transform": "identity"
1968
+ },
1969
+ {
1970
+ "initializer": "flow.flows.2.post.bias",
1971
+ "shape": [
1972
+ 96
1973
+ ],
1974
+ "storage": "direct",
1975
+ "target": "flow.flows.2.post.bias",
1976
+ "transform": "identity"
1977
+ },
1978
+ {
1979
+ "initializer": "flow.flows.4.pre.weight",
1980
+ "shape": [
1981
+ 192,
1982
+ 96,
1983
+ 1
1984
+ ],
1985
+ "storage": "direct",
1986
+ "target": "flow.flows.4.pre.weight",
1987
+ "transform": "identity"
1988
+ },
1989
+ {
1990
+ "initializer": "flow.flows.4.pre.bias",
1991
+ "shape": [
1992
+ 192
1993
+ ],
1994
+ "storage": "direct",
1995
+ "target": "flow.flows.4.pre.bias",
1996
+ "transform": "identity"
1997
+ },
1998
+ {
1999
+ "initializer": "flow.flows.4.enc.in_layers.0.bias",
2000
+ "shape": [
2001
+ 384
2002
+ ],
2003
+ "storage": "direct",
2004
+ "target": "flow.flows.4.enc.in_layers.0.bias",
2005
+ "transform": "identity"
2006
+ },
2007
+ {
2008
+ "initializer": "flow.flows.4.enc.in_layers.0.weight",
2009
+ "shape": [
2010
+ 384,
2011
+ 192,
2012
+ 5
2013
+ ],
2014
+ "storage": "weight_norm_reconstructed",
2015
+ "target": "flow.flows.4.enc.in_layers.0.weight",
2016
+ "transform": "identity"
2017
+ },
2018
+ {
2019
+ "initializer": "flow.flows.4.enc.in_layers.1.bias",
2020
+ "shape": [
2021
+ 384
2022
+ ],
2023
+ "storage": "direct",
2024
+ "target": "flow.flows.4.enc.in_layers.1.bias",
2025
+ "transform": "identity"
2026
+ },
2027
+ {
2028
+ "initializer": "flow.flows.4.enc.in_layers.1.weight",
2029
+ "shape": [
2030
+ 384,
2031
+ 192,
2032
+ 5
2033
+ ],
2034
+ "storage": "weight_norm_reconstructed",
2035
+ "target": "flow.flows.4.enc.in_layers.1.weight",
2036
+ "transform": "identity"
2037
+ },
2038
+ {
2039
+ "initializer": "flow.flows.4.enc.in_layers.2.bias",
2040
+ "shape": [
2041
+ 384
2042
+ ],
2043
+ "storage": "direct",
2044
+ "target": "flow.flows.4.enc.in_layers.2.bias",
2045
+ "transform": "identity"
2046
+ },
2047
+ {
2048
+ "initializer": "flow.flows.4.enc.in_layers.2.weight",
2049
+ "shape": [
2050
+ 384,
2051
+ 192,
2052
+ 5
2053
+ ],
2054
+ "storage": "weight_norm_reconstructed",
2055
+ "target": "flow.flows.4.enc.in_layers.2.weight",
2056
+ "transform": "identity"
2057
+ },
2058
+ {
2059
+ "initializer": "flow.flows.4.enc.in_layers.3.bias",
2060
+ "shape": [
2061
+ 384
2062
+ ],
2063
+ "storage": "direct",
2064
+ "target": "flow.flows.4.enc.in_layers.3.bias",
2065
+ "transform": "identity"
2066
+ },
2067
+ {
2068
+ "initializer": "flow.flows.4.enc.in_layers.3.weight",
2069
+ "shape": [
2070
+ 384,
2071
+ 192,
2072
+ 5
2073
+ ],
2074
+ "storage": "weight_norm_reconstructed",
2075
+ "target": "flow.flows.4.enc.in_layers.3.weight",
2076
+ "transform": "identity"
2077
+ },
2078
+ {
2079
+ "initializer": "flow.flows.4.enc.res_skip_layers.0.bias",
2080
+ "shape": [
2081
+ 384
2082
+ ],
2083
+ "storage": "direct",
2084
+ "target": "flow.flows.4.enc.res_skip_layers.0.bias",
2085
+ "transform": "identity"
2086
+ },
2087
+ {
2088
+ "initializer": "flow.flows.4.enc.res_skip_layers.0.weight",
2089
+ "shape": [
2090
+ 384,
2091
+ 192,
2092
+ 1
2093
+ ],
2094
+ "storage": "weight_norm_reconstructed",
2095
+ "target": "flow.flows.4.enc.res_skip_layers.0.weight",
2096
+ "transform": "identity"
2097
+ },
2098
+ {
2099
+ "initializer": "flow.flows.4.enc.res_skip_layers.1.bias",
2100
+ "shape": [
2101
+ 384
2102
+ ],
2103
+ "storage": "direct",
2104
+ "target": "flow.flows.4.enc.res_skip_layers.1.bias",
2105
+ "transform": "identity"
2106
+ },
2107
+ {
2108
+ "initializer": "flow.flows.4.enc.res_skip_layers.1.weight",
2109
+ "shape": [
2110
+ 384,
2111
+ 192,
2112
+ 1
2113
+ ],
2114
+ "storage": "weight_norm_reconstructed",
2115
+ "target": "flow.flows.4.enc.res_skip_layers.1.weight",
2116
+ "transform": "identity"
2117
+ },
2118
+ {
2119
+ "initializer": "flow.flows.4.enc.res_skip_layers.2.bias",
2120
+ "shape": [
2121
+ 384
2122
+ ],
2123
+ "storage": "direct",
2124
+ "target": "flow.flows.4.enc.res_skip_layers.2.bias",
2125
+ "transform": "identity"
2126
+ },
2127
+ {
2128
+ "initializer": "flow.flows.4.enc.res_skip_layers.2.weight",
2129
+ "shape": [
2130
+ 384,
2131
+ 192,
2132
+ 1
2133
+ ],
2134
+ "storage": "weight_norm_reconstructed",
2135
+ "target": "flow.flows.4.enc.res_skip_layers.2.weight",
2136
+ "transform": "identity"
2137
+ },
2138
+ {
2139
+ "initializer": "flow.flows.4.enc.res_skip_layers.3.bias",
2140
+ "shape": [
2141
+ 192
2142
+ ],
2143
+ "storage": "direct",
2144
+ "target": "flow.flows.4.enc.res_skip_layers.3.bias",
2145
+ "transform": "identity"
2146
+ },
2147
+ {
2148
+ "initializer": "flow.flows.4.enc.res_skip_layers.3.weight",
2149
+ "shape": [
2150
+ 192,
2151
+ 192,
2152
+ 1
2153
+ ],
2154
+ "storage": "weight_norm_reconstructed",
2155
+ "target": "flow.flows.4.enc.res_skip_layers.3.weight",
2156
+ "transform": "identity"
2157
+ },
2158
+ {
2159
+ "initializer": "flow.flows.4.post.weight",
2160
+ "shape": [
2161
+ 96,
2162
+ 192,
2163
+ 1
2164
+ ],
2165
+ "storage": "direct",
2166
+ "target": "flow.flows.4.post.weight",
2167
+ "transform": "identity"
2168
+ },
2169
+ {
2170
+ "initializer": "flow.flows.4.post.bias",
2171
+ "shape": [
2172
+ 96
2173
+ ],
2174
+ "storage": "direct",
2175
+ "target": "flow.flows.4.post.bias",
2176
+ "transform": "identity"
2177
+ },
2178
+ {
2179
+ "initializer": "flow.flows.6.pre.weight",
2180
+ "shape": [
2181
+ 192,
2182
+ 96,
2183
+ 1
2184
+ ],
2185
+ "storage": "direct",
2186
+ "target": "flow.flows.6.pre.weight",
2187
+ "transform": "identity"
2188
+ },
2189
+ {
2190
+ "initializer": "flow.flows.6.pre.bias",
2191
+ "shape": [
2192
+ 192
2193
+ ],
2194
+ "storage": "direct",
2195
+ "target": "flow.flows.6.pre.bias",
2196
+ "transform": "identity"
2197
+ },
2198
+ {
2199
+ "initializer": "flow.flows.6.enc.in_layers.0.bias",
2200
+ "shape": [
2201
+ 384
2202
+ ],
2203
+ "storage": "direct",
2204
+ "target": "flow.flows.6.enc.in_layers.0.bias",
2205
+ "transform": "identity"
2206
+ },
2207
+ {
2208
+ "initializer": "flow.flows.6.enc.in_layers.0.weight",
2209
+ "shape": [
2210
+ 384,
2211
+ 192,
2212
+ 5
2213
+ ],
2214
+ "storage": "weight_norm_reconstructed",
2215
+ "target": "flow.flows.6.enc.in_layers.0.weight",
2216
+ "transform": "identity"
2217
+ },
2218
+ {
2219
+ "initializer": "flow.flows.6.enc.in_layers.1.bias",
2220
+ "shape": [
2221
+ 384
2222
+ ],
2223
+ "storage": "direct",
2224
+ "target": "flow.flows.6.enc.in_layers.1.bias",
2225
+ "transform": "identity"
2226
+ },
2227
+ {
2228
+ "initializer": "flow.flows.6.enc.in_layers.1.weight",
2229
+ "shape": [
2230
+ 384,
2231
+ 192,
2232
+ 5
2233
+ ],
2234
+ "storage": "weight_norm_reconstructed",
2235
+ "target": "flow.flows.6.enc.in_layers.1.weight",
2236
+ "transform": "identity"
2237
+ },
2238
+ {
2239
+ "initializer": "flow.flows.6.enc.in_layers.2.bias",
2240
+ "shape": [
2241
+ 384
2242
+ ],
2243
+ "storage": "direct",
2244
+ "target": "flow.flows.6.enc.in_layers.2.bias",
2245
+ "transform": "identity"
2246
+ },
2247
+ {
2248
+ "initializer": "flow.flows.6.enc.in_layers.2.weight",
2249
+ "shape": [
2250
+ 384,
2251
+ 192,
2252
+ 5
2253
+ ],
2254
+ "storage": "weight_norm_reconstructed",
2255
+ "target": "flow.flows.6.enc.in_layers.2.weight",
2256
+ "transform": "identity"
2257
+ },
2258
+ {
2259
+ "initializer": "flow.flows.6.enc.in_layers.3.bias",
2260
+ "shape": [
2261
+ 384
2262
+ ],
2263
+ "storage": "direct",
2264
+ "target": "flow.flows.6.enc.in_layers.3.bias",
2265
+ "transform": "identity"
2266
+ },
2267
+ {
2268
+ "initializer": "flow.flows.6.enc.in_layers.3.weight",
2269
+ "shape": [
2270
+ 384,
2271
+ 192,
2272
+ 5
2273
+ ],
2274
+ "storage": "weight_norm_reconstructed",
2275
+ "target": "flow.flows.6.enc.in_layers.3.weight",
2276
+ "transform": "identity"
2277
+ },
2278
+ {
2279
+ "initializer": "flow.flows.6.enc.res_skip_layers.0.bias",
2280
+ "shape": [
2281
+ 384
2282
+ ],
2283
+ "storage": "direct",
2284
+ "target": "flow.flows.6.enc.res_skip_layers.0.bias",
2285
+ "transform": "identity"
2286
+ },
2287
+ {
2288
+ "initializer": "flow.flows.6.enc.res_skip_layers.0.weight",
2289
+ "shape": [
2290
+ 384,
2291
+ 192,
2292
+ 1
2293
+ ],
2294
+ "storage": "weight_norm_reconstructed",
2295
+ "target": "flow.flows.6.enc.res_skip_layers.0.weight",
2296
+ "transform": "identity"
2297
+ },
2298
+ {
2299
+ "initializer": "flow.flows.6.enc.res_skip_layers.1.bias",
2300
+ "shape": [
2301
+ 384
2302
+ ],
2303
+ "storage": "direct",
2304
+ "target": "flow.flows.6.enc.res_skip_layers.1.bias",
2305
+ "transform": "identity"
2306
+ },
2307
+ {
2308
+ "initializer": "flow.flows.6.enc.res_skip_layers.1.weight",
2309
+ "shape": [
2310
+ 384,
2311
+ 192,
2312
+ 1
2313
+ ],
2314
+ "storage": "weight_norm_reconstructed",
2315
+ "target": "flow.flows.6.enc.res_skip_layers.1.weight",
2316
+ "transform": "identity"
2317
+ },
2318
+ {
2319
+ "initializer": "flow.flows.6.enc.res_skip_layers.2.bias",
2320
+ "shape": [
2321
+ 384
2322
+ ],
2323
+ "storage": "direct",
2324
+ "target": "flow.flows.6.enc.res_skip_layers.2.bias",
2325
+ "transform": "identity"
2326
+ },
2327
+ {
2328
+ "initializer": "flow.flows.6.enc.res_skip_layers.2.weight",
2329
+ "shape": [
2330
+ 384,
2331
+ 192,
2332
+ 1
2333
+ ],
2334
+ "storage": "weight_norm_reconstructed",
2335
+ "target": "flow.flows.6.enc.res_skip_layers.2.weight",
2336
+ "transform": "identity"
2337
+ },
2338
+ {
2339
+ "initializer": "flow.flows.6.enc.res_skip_layers.3.bias",
2340
+ "shape": [
2341
+ 192
2342
+ ],
2343
+ "storage": "direct",
2344
+ "target": "flow.flows.6.enc.res_skip_layers.3.bias",
2345
+ "transform": "identity"
2346
+ },
2347
+ {
2348
+ "initializer": "flow.flows.6.enc.res_skip_layers.3.weight",
2349
+ "shape": [
2350
+ 192,
2351
+ 192,
2352
+ 1
2353
+ ],
2354
+ "storage": "weight_norm_reconstructed",
2355
+ "target": "flow.flows.6.enc.res_skip_layers.3.weight",
2356
+ "transform": "identity"
2357
+ },
2358
+ {
2359
+ "initializer": "flow.flows.6.post.weight",
2360
+ "shape": [
2361
+ 96,
2362
+ 192,
2363
+ 1
2364
+ ],
2365
+ "storage": "direct",
2366
+ "target": "flow.flows.6.post.weight",
2367
+ "transform": "identity"
2368
+ },
2369
+ {
2370
+ "initializer": "flow.flows.6.post.bias",
2371
+ "shape": [
2372
+ 96
2373
+ ],
2374
+ "storage": "direct",
2375
+ "target": "flow.flows.6.post.bias",
2376
+ "transform": "identity"
2377
+ },
2378
+ {
2379
+ "initializer": "dp.flows.0.m",
2380
+ "shape": [
2381
+ 2,
2382
+ 1
2383
+ ],
2384
+ "storage": "direct",
2385
+ "target": "dp.flows.0.m",
2386
+ "transform": "identity"
2387
+ },
2388
+ {
2389
+ "initializer": "dp.flows.3.pre.weight",
2390
+ "shape": [
2391
+ 192,
2392
+ 1,
2393
+ 1
2394
+ ],
2395
+ "storage": "direct",
2396
+ "target": "dp.flows.3.pre.weight",
2397
+ "transform": "identity"
2398
+ },
2399
+ {
2400
+ "initializer": "dp.flows.3.pre.bias",
2401
+ "shape": [
2402
+ 192
2403
+ ],
2404
+ "storage": "direct",
2405
+ "target": "dp.flows.3.pre.bias",
2406
+ "transform": "identity"
2407
+ },
2408
+ {
2409
+ "initializer": "dp.flows.3.convs.convs_sep.0.weight",
2410
+ "shape": [
2411
+ 192,
2412
+ 1,
2413
+ 3
2414
+ ],
2415
+ "storage": "direct",
2416
+ "target": "dp.flows.3.convs.convs_sep.0.weight",
2417
+ "transform": "identity"
2418
+ },
2419
+ {
2420
+ "initializer": "dp.flows.3.convs.convs_sep.0.bias",
2421
+ "shape": [
2422
+ 192
2423
+ ],
2424
+ "storage": "direct",
2425
+ "target": "dp.flows.3.convs.convs_sep.0.bias",
2426
+ "transform": "identity"
2427
+ },
2428
+ {
2429
+ "initializer": "dp.flows.3.convs.convs_sep.1.weight",
2430
+ "shape": [
2431
+ 192,
2432
+ 1,
2433
+ 3
2434
+ ],
2435
+ "storage": "direct",
2436
+ "target": "dp.flows.3.convs.convs_sep.1.weight",
2437
+ "transform": "identity"
2438
+ },
2439
+ {
2440
+ "initializer": "dp.flows.3.convs.convs_sep.1.bias",
2441
+ "shape": [
2442
+ 192
2443
+ ],
2444
+ "storage": "direct",
2445
+ "target": "dp.flows.3.convs.convs_sep.1.bias",
2446
+ "transform": "identity"
2447
+ },
2448
+ {
2449
+ "initializer": "dp.flows.3.convs.convs_sep.2.weight",
2450
+ "shape": [
2451
+ 192,
2452
+ 1,
2453
+ 3
2454
+ ],
2455
+ "storage": "direct",
2456
+ "target": "dp.flows.3.convs.convs_sep.2.weight",
2457
+ "transform": "identity"
2458
+ },
2459
+ {
2460
+ "initializer": "dp.flows.3.convs.convs_sep.2.bias",
2461
+ "shape": [
2462
+ 192
2463
+ ],
2464
+ "storage": "direct",
2465
+ "target": "dp.flows.3.convs.convs_sep.2.bias",
2466
+ "transform": "identity"
2467
+ },
2468
+ {
2469
+ "initializer": "dp.flows.3.convs.convs_1x1.0.weight",
2470
+ "shape": [
2471
+ 192,
2472
+ 192,
2473
+ 1
2474
+ ],
2475
+ "storage": "direct",
2476
+ "target": "dp.flows.3.convs.convs_1x1.0.weight",
2477
+ "transform": "identity"
2478
+ },
2479
+ {
2480
+ "initializer": "dp.flows.3.convs.convs_1x1.0.bias",
2481
+ "shape": [
2482
+ 192
2483
+ ],
2484
+ "storage": "direct",
2485
+ "target": "dp.flows.3.convs.convs_1x1.0.bias",
2486
+ "transform": "identity"
2487
+ },
2488
+ {
2489
+ "initializer": "dp.flows.3.convs.convs_1x1.1.weight",
2490
+ "shape": [
2491
+ 192,
2492
+ 192,
2493
+ 1
2494
+ ],
2495
+ "storage": "direct",
2496
+ "target": "dp.flows.3.convs.convs_1x1.1.weight",
2497
+ "transform": "identity"
2498
+ },
2499
+ {
2500
+ "initializer": "dp.flows.3.convs.convs_1x1.1.bias",
2501
+ "shape": [
2502
+ 192
2503
+ ],
2504
+ "storage": "direct",
2505
+ "target": "dp.flows.3.convs.convs_1x1.1.bias",
2506
+ "transform": "identity"
2507
+ },
2508
+ {
2509
+ "initializer": "dp.flows.3.convs.convs_1x1.2.weight",
2510
+ "shape": [
2511
+ 192,
2512
+ 192,
2513
+ 1
2514
+ ],
2515
+ "storage": "direct",
2516
+ "target": "dp.flows.3.convs.convs_1x1.2.weight",
2517
+ "transform": "identity"
2518
+ },
2519
+ {
2520
+ "initializer": "dp.flows.3.convs.convs_1x1.2.bias",
2521
+ "shape": [
2522
+ 192
2523
+ ],
2524
+ "storage": "direct",
2525
+ "target": "dp.flows.3.convs.convs_1x1.2.bias",
2526
+ "transform": "identity"
2527
+ },
2528
+ {
2529
+ "initializer": "dp.flows.3.convs.norms_1.0.gamma",
2530
+ "shape": [
2531
+ 192
2532
+ ],
2533
+ "storage": "direct",
2534
+ "target": "dp.flows.3.convs.norms_1.0.gamma",
2535
+ "transform": "identity"
2536
+ },
2537
+ {
2538
+ "initializer": "dp.flows.3.convs.norms_1.0.beta",
2539
+ "shape": [
2540
+ 192
2541
+ ],
2542
+ "storage": "direct",
2543
+ "target": "dp.flows.3.convs.norms_1.0.beta",
2544
+ "transform": "identity"
2545
+ },
2546
+ {
2547
+ "initializer": "dp.flows.3.convs.norms_1.1.gamma",
2548
+ "shape": [
2549
+ 192
2550
+ ],
2551
+ "storage": "direct",
2552
+ "target": "dp.flows.3.convs.norms_1.1.gamma",
2553
+ "transform": "identity"
2554
+ },
2555
+ {
2556
+ "initializer": "dp.flows.3.convs.norms_1.1.beta",
2557
+ "shape": [
2558
+ 192
2559
+ ],
2560
+ "storage": "direct",
2561
+ "target": "dp.flows.3.convs.norms_1.1.beta",
2562
+ "transform": "identity"
2563
+ },
2564
+ {
2565
+ "initializer": "dp.flows.3.convs.norms_1.2.gamma",
2566
+ "shape": [
2567
+ 192
2568
+ ],
2569
+ "storage": "direct",
2570
+ "target": "dp.flows.3.convs.norms_1.2.gamma",
2571
+ "transform": "identity"
2572
+ },
2573
+ {
2574
+ "initializer": "dp.flows.3.convs.norms_1.2.beta",
2575
+ "shape": [
2576
+ 192
2577
+ ],
2578
+ "storage": "direct",
2579
+ "target": "dp.flows.3.convs.norms_1.2.beta",
2580
+ "transform": "identity"
2581
+ },
2582
+ {
2583
+ "initializer": "dp.flows.3.convs.norms_2.0.gamma",
2584
+ "shape": [
2585
+ 192
2586
+ ],
2587
+ "storage": "direct",
2588
+ "target": "dp.flows.3.convs.norms_2.0.gamma",
2589
+ "transform": "identity"
2590
+ },
2591
+ {
2592
+ "initializer": "dp.flows.3.convs.norms_2.0.beta",
2593
+ "shape": [
2594
+ 192
2595
+ ],
2596
+ "storage": "direct",
2597
+ "target": "dp.flows.3.convs.norms_2.0.beta",
2598
+ "transform": "identity"
2599
+ },
2600
+ {
2601
+ "initializer": "dp.flows.3.convs.norms_2.1.gamma",
2602
+ "shape": [
2603
+ 192
2604
+ ],
2605
+ "storage": "direct",
2606
+ "target": "dp.flows.3.convs.norms_2.1.gamma",
2607
+ "transform": "identity"
2608
+ },
2609
+ {
2610
+ "initializer": "dp.flows.3.convs.norms_2.1.beta",
2611
+ "shape": [
2612
+ 192
2613
+ ],
2614
+ "storage": "direct",
2615
+ "target": "dp.flows.3.convs.norms_2.1.beta",
2616
+ "transform": "identity"
2617
+ },
2618
+ {
2619
+ "initializer": "dp.flows.3.convs.norms_2.2.gamma",
2620
+ "shape": [
2621
+ 192
2622
+ ],
2623
+ "storage": "direct",
2624
+ "target": "dp.flows.3.convs.norms_2.2.gamma",
2625
+ "transform": "identity"
2626
+ },
2627
+ {
2628
+ "initializer": "dp.flows.3.convs.norms_2.2.beta",
2629
+ "shape": [
2630
+ 192
2631
+ ],
2632
+ "storage": "direct",
2633
+ "target": "dp.flows.3.convs.norms_2.2.beta",
2634
+ "transform": "identity"
2635
+ },
2636
+ {
2637
+ "initializer": "dp.flows.3.proj.weight",
2638
+ "shape": [
2639
+ 29,
2640
+ 192,
2641
+ 1
2642
+ ],
2643
+ "storage": "direct",
2644
+ "target": "dp.flows.3.proj.weight",
2645
+ "transform": "identity"
2646
+ },
2647
+ {
2648
+ "initializer": "dp.flows.3.proj.bias",
2649
+ "shape": [
2650
+ 29
2651
+ ],
2652
+ "storage": "direct",
2653
+ "target": "dp.flows.3.proj.bias",
2654
+ "transform": "identity"
2655
+ },
2656
+ {
2657
+ "initializer": "dp.flows.5.pre.weight",
2658
+ "shape": [
2659
+ 192,
2660
+ 1,
2661
+ 1
2662
+ ],
2663
+ "storage": "direct",
2664
+ "target": "dp.flows.5.pre.weight",
2665
+ "transform": "identity"
2666
+ },
2667
+ {
2668
+ "initializer": "dp.flows.5.pre.bias",
2669
+ "shape": [
2670
+ 192
2671
+ ],
2672
+ "storage": "direct",
2673
+ "target": "dp.flows.5.pre.bias",
2674
+ "transform": "identity"
2675
+ },
2676
+ {
2677
+ "initializer": "dp.flows.5.convs.convs_sep.0.weight",
2678
+ "shape": [
2679
+ 192,
2680
+ 1,
2681
+ 3
2682
+ ],
2683
+ "storage": "direct",
2684
+ "target": "dp.flows.5.convs.convs_sep.0.weight",
2685
+ "transform": "identity"
2686
+ },
2687
+ {
2688
+ "initializer": "dp.flows.5.convs.convs_sep.0.bias",
2689
+ "shape": [
2690
+ 192
2691
+ ],
2692
+ "storage": "direct",
2693
+ "target": "dp.flows.5.convs.convs_sep.0.bias",
2694
+ "transform": "identity"
2695
+ },
2696
+ {
2697
+ "initializer": "dp.flows.5.convs.convs_sep.1.weight",
2698
+ "shape": [
2699
+ 192,
2700
+ 1,
2701
+ 3
2702
+ ],
2703
+ "storage": "direct",
2704
+ "target": "dp.flows.5.convs.convs_sep.1.weight",
2705
+ "transform": "identity"
2706
+ },
2707
+ {
2708
+ "initializer": "dp.flows.5.convs.convs_sep.1.bias",
2709
+ "shape": [
2710
+ 192
2711
+ ],
2712
+ "storage": "direct",
2713
+ "target": "dp.flows.5.convs.convs_sep.1.bias",
2714
+ "transform": "identity"
2715
+ },
2716
+ {
2717
+ "initializer": "dp.flows.5.convs.convs_sep.2.weight",
2718
+ "shape": [
2719
+ 192,
2720
+ 1,
2721
+ 3
2722
+ ],
2723
+ "storage": "direct",
2724
+ "target": "dp.flows.5.convs.convs_sep.2.weight",
2725
+ "transform": "identity"
2726
+ },
2727
+ {
2728
+ "initializer": "dp.flows.5.convs.convs_sep.2.bias",
2729
+ "shape": [
2730
+ 192
2731
+ ],
2732
+ "storage": "direct",
2733
+ "target": "dp.flows.5.convs.convs_sep.2.bias",
2734
+ "transform": "identity"
2735
+ },
2736
+ {
2737
+ "initializer": "dp.flows.5.convs.convs_1x1.0.weight",
2738
+ "shape": [
2739
+ 192,
2740
+ 192,
2741
+ 1
2742
+ ],
2743
+ "storage": "direct",
2744
+ "target": "dp.flows.5.convs.convs_1x1.0.weight",
2745
+ "transform": "identity"
2746
+ },
2747
+ {
2748
+ "initializer": "dp.flows.5.convs.convs_1x1.0.bias",
2749
+ "shape": [
2750
+ 192
2751
+ ],
2752
+ "storage": "direct",
2753
+ "target": "dp.flows.5.convs.convs_1x1.0.bias",
2754
+ "transform": "identity"
2755
+ },
2756
+ {
2757
+ "initializer": "dp.flows.5.convs.convs_1x1.1.weight",
2758
+ "shape": [
2759
+ 192,
2760
+ 192,
2761
+ 1
2762
+ ],
2763
+ "storage": "direct",
2764
+ "target": "dp.flows.5.convs.convs_1x1.1.weight",
2765
+ "transform": "identity"
2766
+ },
2767
+ {
2768
+ "initializer": "dp.flows.5.convs.convs_1x1.1.bias",
2769
+ "shape": [
2770
+ 192
2771
+ ],
2772
+ "storage": "direct",
2773
+ "target": "dp.flows.5.convs.convs_1x1.1.bias",
2774
+ "transform": "identity"
2775
+ },
2776
+ {
2777
+ "initializer": "dp.flows.5.convs.convs_1x1.2.weight",
2778
+ "shape": [
2779
+ 192,
2780
+ 192,
2781
+ 1
2782
+ ],
2783
+ "storage": "direct",
2784
+ "target": "dp.flows.5.convs.convs_1x1.2.weight",
2785
+ "transform": "identity"
2786
+ },
2787
+ {
2788
+ "initializer": "dp.flows.5.convs.convs_1x1.2.bias",
2789
+ "shape": [
2790
+ 192
2791
+ ],
2792
+ "storage": "direct",
2793
+ "target": "dp.flows.5.convs.convs_1x1.2.bias",
2794
+ "transform": "identity"
2795
+ },
2796
+ {
2797
+ "initializer": "dp.flows.5.convs.norms_1.0.gamma",
2798
+ "shape": [
2799
+ 192
2800
+ ],
2801
+ "storage": "direct",
2802
+ "target": "dp.flows.5.convs.norms_1.0.gamma",
2803
+ "transform": "identity"
2804
+ },
2805
+ {
2806
+ "initializer": "dp.flows.5.convs.norms_1.0.beta",
2807
+ "shape": [
2808
+ 192
2809
+ ],
2810
+ "storage": "direct",
2811
+ "target": "dp.flows.5.convs.norms_1.0.beta",
2812
+ "transform": "identity"
2813
+ },
2814
+ {
2815
+ "initializer": "dp.flows.5.convs.norms_1.1.gamma",
2816
+ "shape": [
2817
+ 192
2818
+ ],
2819
+ "storage": "direct",
2820
+ "target": "dp.flows.5.convs.norms_1.1.gamma",
2821
+ "transform": "identity"
2822
+ },
2823
+ {
2824
+ "initializer": "dp.flows.5.convs.norms_1.1.beta",
2825
+ "shape": [
2826
+ 192
2827
+ ],
2828
+ "storage": "direct",
2829
+ "target": "dp.flows.5.convs.norms_1.1.beta",
2830
+ "transform": "identity"
2831
+ },
2832
+ {
2833
+ "initializer": "dp.flows.5.convs.norms_1.2.gamma",
2834
+ "shape": [
2835
+ 192
2836
+ ],
2837
+ "storage": "direct",
2838
+ "target": "dp.flows.5.convs.norms_1.2.gamma",
2839
+ "transform": "identity"
2840
+ },
2841
+ {
2842
+ "initializer": "dp.flows.5.convs.norms_1.2.beta",
2843
+ "shape": [
2844
+ 192
2845
+ ],
2846
+ "storage": "direct",
2847
+ "target": "dp.flows.5.convs.norms_1.2.beta",
2848
+ "transform": "identity"
2849
+ },
2850
+ {
2851
+ "initializer": "dp.flows.5.convs.norms_2.0.gamma",
2852
+ "shape": [
2853
+ 192
2854
+ ],
2855
+ "storage": "direct",
2856
+ "target": "dp.flows.5.convs.norms_2.0.gamma",
2857
+ "transform": "identity"
2858
+ },
2859
+ {
2860
+ "initializer": "dp.flows.5.convs.norms_2.0.beta",
2861
+ "shape": [
2862
+ 192
2863
+ ],
2864
+ "storage": "direct",
2865
+ "target": "dp.flows.5.convs.norms_2.0.beta",
2866
+ "transform": "identity"
2867
+ },
2868
+ {
2869
+ "initializer": "dp.flows.5.convs.norms_2.1.gamma",
2870
+ "shape": [
2871
+ 192
2872
+ ],
2873
+ "storage": "direct",
2874
+ "target": "dp.flows.5.convs.norms_2.1.gamma",
2875
+ "transform": "identity"
2876
+ },
2877
+ {
2878
+ "initializer": "dp.flows.5.convs.norms_2.1.beta",
2879
+ "shape": [
2880
+ 192
2881
+ ],
2882
+ "storage": "direct",
2883
+ "target": "dp.flows.5.convs.norms_2.1.beta",
2884
+ "transform": "identity"
2885
+ },
2886
+ {
2887
+ "initializer": "dp.flows.5.convs.norms_2.2.gamma",
2888
+ "shape": [
2889
+ 192
2890
+ ],
2891
+ "storage": "direct",
2892
+ "target": "dp.flows.5.convs.norms_2.2.gamma",
2893
+ "transform": "identity"
2894
+ },
2895
+ {
2896
+ "initializer": "dp.flows.5.convs.norms_2.2.beta",
2897
+ "shape": [
2898
+ 192
2899
+ ],
2900
+ "storage": "direct",
2901
+ "target": "dp.flows.5.convs.norms_2.2.beta",
2902
+ "transform": "identity"
2903
+ },
2904
+ {
2905
+ "initializer": "dp.flows.5.proj.weight",
2906
+ "shape": [
2907
+ 29,
2908
+ 192,
2909
+ 1
2910
+ ],
2911
+ "storage": "direct",
2912
+ "target": "dp.flows.5.proj.weight",
2913
+ "transform": "identity"
2914
+ },
2915
+ {
2916
+ "initializer": "dp.flows.5.proj.bias",
2917
+ "shape": [
2918
+ 29
2919
+ ],
2920
+ "storage": "direct",
2921
+ "target": "dp.flows.5.proj.bias",
2922
+ "transform": "identity"
2923
+ },
2924
+ {
2925
+ "initializer": "dp.flows.7.pre.weight",
2926
+ "shape": [
2927
+ 192,
2928
+ 1,
2929
+ 1
2930
+ ],
2931
+ "storage": "direct",
2932
+ "target": "dp.flows.7.pre.weight",
2933
+ "transform": "identity"
2934
+ },
2935
+ {
2936
+ "initializer": "dp.flows.7.pre.bias",
2937
+ "shape": [
2938
+ 192
2939
+ ],
2940
+ "storage": "direct",
2941
+ "target": "dp.flows.7.pre.bias",
2942
+ "transform": "identity"
2943
+ },
2944
+ {
2945
+ "initializer": "dp.flows.7.convs.convs_sep.0.weight",
2946
+ "shape": [
2947
+ 192,
2948
+ 1,
2949
+ 3
2950
+ ],
2951
+ "storage": "direct",
2952
+ "target": "dp.flows.7.convs.convs_sep.0.weight",
2953
+ "transform": "identity"
2954
+ },
2955
+ {
2956
+ "initializer": "dp.flows.7.convs.convs_sep.0.bias",
2957
+ "shape": [
2958
+ 192
2959
+ ],
2960
+ "storage": "direct",
2961
+ "target": "dp.flows.7.convs.convs_sep.0.bias",
2962
+ "transform": "identity"
2963
+ },
2964
+ {
2965
+ "initializer": "dp.flows.7.convs.convs_sep.1.weight",
2966
+ "shape": [
2967
+ 192,
2968
+ 1,
2969
+ 3
2970
+ ],
2971
+ "storage": "direct",
2972
+ "target": "dp.flows.7.convs.convs_sep.1.weight",
2973
+ "transform": "identity"
2974
+ },
2975
+ {
2976
+ "initializer": "dp.flows.7.convs.convs_sep.1.bias",
2977
+ "shape": [
2978
+ 192
2979
+ ],
2980
+ "storage": "direct",
2981
+ "target": "dp.flows.7.convs.convs_sep.1.bias",
2982
+ "transform": "identity"
2983
+ },
2984
+ {
2985
+ "initializer": "dp.flows.7.convs.convs_sep.2.weight",
2986
+ "shape": [
2987
+ 192,
2988
+ 1,
2989
+ 3
2990
+ ],
2991
+ "storage": "direct",
2992
+ "target": "dp.flows.7.convs.convs_sep.2.weight",
2993
+ "transform": "identity"
2994
+ },
2995
+ {
2996
+ "initializer": "dp.flows.7.convs.convs_sep.2.bias",
2997
+ "shape": [
2998
+ 192
2999
+ ],
3000
+ "storage": "direct",
3001
+ "target": "dp.flows.7.convs.convs_sep.2.bias",
3002
+ "transform": "identity"
3003
+ },
3004
+ {
3005
+ "initializer": "dp.flows.7.convs.convs_1x1.0.weight",
3006
+ "shape": [
3007
+ 192,
3008
+ 192,
3009
+ 1
3010
+ ],
3011
+ "storage": "direct",
3012
+ "target": "dp.flows.7.convs.convs_1x1.0.weight",
3013
+ "transform": "identity"
3014
+ },
3015
+ {
3016
+ "initializer": "dp.flows.7.convs.convs_1x1.0.bias",
3017
+ "shape": [
3018
+ 192
3019
+ ],
3020
+ "storage": "direct",
3021
+ "target": "dp.flows.7.convs.convs_1x1.0.bias",
3022
+ "transform": "identity"
3023
+ },
3024
+ {
3025
+ "initializer": "dp.flows.7.convs.convs_1x1.1.weight",
3026
+ "shape": [
3027
+ 192,
3028
+ 192,
3029
+ 1
3030
+ ],
3031
+ "storage": "direct",
3032
+ "target": "dp.flows.7.convs.convs_1x1.1.weight",
3033
+ "transform": "identity"
3034
+ },
3035
+ {
3036
+ "initializer": "dp.flows.7.convs.convs_1x1.1.bias",
3037
+ "shape": [
3038
+ 192
3039
+ ],
3040
+ "storage": "direct",
3041
+ "target": "dp.flows.7.convs.convs_1x1.1.bias",
3042
+ "transform": "identity"
3043
+ },
3044
+ {
3045
+ "initializer": "dp.flows.7.convs.convs_1x1.2.weight",
3046
+ "shape": [
3047
+ 192,
3048
+ 192,
3049
+ 1
3050
+ ],
3051
+ "storage": "direct",
3052
+ "target": "dp.flows.7.convs.convs_1x1.2.weight",
3053
+ "transform": "identity"
3054
+ },
3055
+ {
3056
+ "initializer": "dp.flows.7.convs.convs_1x1.2.bias",
3057
+ "shape": [
3058
+ 192
3059
+ ],
3060
+ "storage": "direct",
3061
+ "target": "dp.flows.7.convs.convs_1x1.2.bias",
3062
+ "transform": "identity"
3063
+ },
3064
+ {
3065
+ "initializer": "dp.flows.7.convs.norms_1.0.gamma",
3066
+ "shape": [
3067
+ 192
3068
+ ],
3069
+ "storage": "direct",
3070
+ "target": "dp.flows.7.convs.norms_1.0.gamma",
3071
+ "transform": "identity"
3072
+ },
3073
+ {
3074
+ "initializer": "dp.flows.7.convs.norms_1.0.beta",
3075
+ "shape": [
3076
+ 192
3077
+ ],
3078
+ "storage": "direct",
3079
+ "target": "dp.flows.7.convs.norms_1.0.beta",
3080
+ "transform": "identity"
3081
+ },
3082
+ {
3083
+ "initializer": "dp.flows.7.convs.norms_1.1.gamma",
3084
+ "shape": [
3085
+ 192
3086
+ ],
3087
+ "storage": "direct",
3088
+ "target": "dp.flows.7.convs.norms_1.1.gamma",
3089
+ "transform": "identity"
3090
+ },
3091
+ {
3092
+ "initializer": "dp.flows.7.convs.norms_1.1.beta",
3093
+ "shape": [
3094
+ 192
3095
+ ],
3096
+ "storage": "direct",
3097
+ "target": "dp.flows.7.convs.norms_1.1.beta",
3098
+ "transform": "identity"
3099
+ },
3100
+ {
3101
+ "initializer": "dp.flows.7.convs.norms_1.2.gamma",
3102
+ "shape": [
3103
+ 192
3104
+ ],
3105
+ "storage": "direct",
3106
+ "target": "dp.flows.7.convs.norms_1.2.gamma",
3107
+ "transform": "identity"
3108
+ },
3109
+ {
3110
+ "initializer": "dp.flows.7.convs.norms_1.2.beta",
3111
+ "shape": [
3112
+ 192
3113
+ ],
3114
+ "storage": "direct",
3115
+ "target": "dp.flows.7.convs.norms_1.2.beta",
3116
+ "transform": "identity"
3117
+ },
3118
+ {
3119
+ "initializer": "dp.flows.7.convs.norms_2.0.gamma",
3120
+ "shape": [
3121
+ 192
3122
+ ],
3123
+ "storage": "direct",
3124
+ "target": "dp.flows.7.convs.norms_2.0.gamma",
3125
+ "transform": "identity"
3126
+ },
3127
+ {
3128
+ "initializer": "dp.flows.7.convs.norms_2.0.beta",
3129
+ "shape": [
3130
+ 192
3131
+ ],
3132
+ "storage": "direct",
3133
+ "target": "dp.flows.7.convs.norms_2.0.beta",
3134
+ "transform": "identity"
3135
+ },
3136
+ {
3137
+ "initializer": "dp.flows.7.convs.norms_2.1.gamma",
3138
+ "shape": [
3139
+ 192
3140
+ ],
3141
+ "storage": "direct",
3142
+ "target": "dp.flows.7.convs.norms_2.1.gamma",
3143
+ "transform": "identity"
3144
+ },
3145
+ {
3146
+ "initializer": "dp.flows.7.convs.norms_2.1.beta",
3147
+ "shape": [
3148
+ 192
3149
+ ],
3150
+ "storage": "direct",
3151
+ "target": "dp.flows.7.convs.norms_2.1.beta",
3152
+ "transform": "identity"
3153
+ },
3154
+ {
3155
+ "initializer": "dp.flows.7.convs.norms_2.2.gamma",
3156
+ "shape": [
3157
+ 192
3158
+ ],
3159
+ "storage": "direct",
3160
+ "target": "dp.flows.7.convs.norms_2.2.gamma",
3161
+ "transform": "identity"
3162
+ },
3163
+ {
3164
+ "initializer": "dp.flows.7.convs.norms_2.2.beta",
3165
+ "shape": [
3166
+ 192
3167
+ ],
3168
+ "storage": "direct",
3169
+ "target": "dp.flows.7.convs.norms_2.2.beta",
3170
+ "transform": "identity"
3171
+ },
3172
+ {
3173
+ "initializer": "dp.flows.7.proj.weight",
3174
+ "shape": [
3175
+ 29,
3176
+ 192,
3177
+ 1
3178
+ ],
3179
+ "storage": "direct",
3180
+ "target": "dp.flows.7.proj.weight",
3181
+ "transform": "identity"
3182
+ },
3183
+ {
3184
+ "initializer": "dp.flows.7.proj.bias",
3185
+ "shape": [
3186
+ 29
3187
+ ],
3188
+ "storage": "direct",
3189
+ "target": "dp.flows.7.proj.bias",
3190
+ "transform": "identity"
3191
+ },
3192
+ {
3193
+ "initializer": "dp.pre.weight",
3194
+ "shape": [
3195
+ 192,
3196
+ 192,
3197
+ 1
3198
+ ],
3199
+ "storage": "direct",
3200
+ "target": "dp.pre.weight",
3201
+ "transform": "identity"
3202
+ },
3203
+ {
3204
+ "initializer": "dp.pre.bias",
3205
+ "shape": [
3206
+ 192
3207
+ ],
3208
+ "storage": "direct",
3209
+ "target": "dp.pre.bias",
3210
+ "transform": "identity"
3211
+ },
3212
+ {
3213
+ "initializer": "dp.proj.weight",
3214
+ "shape": [
3215
+ 192,
3216
+ 192,
3217
+ 1
3218
+ ],
3219
+ "storage": "direct",
3220
+ "target": "dp.proj.weight",
3221
+ "transform": "identity"
3222
+ },
3223
+ {
3224
+ "initializer": "dp.proj.bias",
3225
+ "shape": [
3226
+ 192
3227
+ ],
3228
+ "storage": "direct",
3229
+ "target": "dp.proj.bias",
3230
+ "transform": "identity"
3231
+ },
3232
+ {
3233
+ "initializer": "dp.convs.convs_sep.0.weight",
3234
+ "shape": [
3235
+ 192,
3236
+ 1,
3237
+ 3
3238
+ ],
3239
+ "storage": "direct",
3240
+ "target": "dp.convs.convs_sep.0.weight",
3241
+ "transform": "identity"
3242
+ },
3243
+ {
3244
+ "initializer": "dp.convs.convs_sep.0.bias",
3245
+ "shape": [
3246
+ 192
3247
+ ],
3248
+ "storage": "direct",
3249
+ "target": "dp.convs.convs_sep.0.bias",
3250
+ "transform": "identity"
3251
+ },
3252
+ {
3253
+ "initializer": "dp.convs.convs_sep.1.weight",
3254
+ "shape": [
3255
+ 192,
3256
+ 1,
3257
+ 3
3258
+ ],
3259
+ "storage": "direct",
3260
+ "target": "dp.convs.convs_sep.1.weight",
3261
+ "transform": "identity"
3262
+ },
3263
+ {
3264
+ "initializer": "dp.convs.convs_sep.1.bias",
3265
+ "shape": [
3266
+ 192
3267
+ ],
3268
+ "storage": "direct",
3269
+ "target": "dp.convs.convs_sep.1.bias",
3270
+ "transform": "identity"
3271
+ },
3272
+ {
3273
+ "initializer": "dp.convs.convs_sep.2.weight",
3274
+ "shape": [
3275
+ 192,
3276
+ 1,
3277
+ 3
3278
+ ],
3279
+ "storage": "direct",
3280
+ "target": "dp.convs.convs_sep.2.weight",
3281
+ "transform": "identity"
3282
+ },
3283
+ {
3284
+ "initializer": "dp.convs.convs_sep.2.bias",
3285
+ "shape": [
3286
+ 192
3287
+ ],
3288
+ "storage": "direct",
3289
+ "target": "dp.convs.convs_sep.2.bias",
3290
+ "transform": "identity"
3291
+ },
3292
+ {
3293
+ "initializer": "dp.convs.convs_1x1.0.weight",
3294
+ "shape": [
3295
+ 192,
3296
+ 192,
3297
+ 1
3298
+ ],
3299
+ "storage": "direct",
3300
+ "target": "dp.convs.convs_1x1.0.weight",
3301
+ "transform": "identity"
3302
+ },
3303
+ {
3304
+ "initializer": "dp.convs.convs_1x1.0.bias",
3305
+ "shape": [
3306
+ 192
3307
+ ],
3308
+ "storage": "direct",
3309
+ "target": "dp.convs.convs_1x1.0.bias",
3310
+ "transform": "identity"
3311
+ },
3312
+ {
3313
+ "initializer": "dp.convs.convs_1x1.1.weight",
3314
+ "shape": [
3315
+ 192,
3316
+ 192,
3317
+ 1
3318
+ ],
3319
+ "storage": "direct",
3320
+ "target": "dp.convs.convs_1x1.1.weight",
3321
+ "transform": "identity"
3322
+ },
3323
+ {
3324
+ "initializer": "dp.convs.convs_1x1.1.bias",
3325
+ "shape": [
3326
+ 192
3327
+ ],
3328
+ "storage": "direct",
3329
+ "target": "dp.convs.convs_1x1.1.bias",
3330
+ "transform": "identity"
3331
+ },
3332
+ {
3333
+ "initializer": "dp.convs.convs_1x1.2.weight",
3334
+ "shape": [
3335
+ 192,
3336
+ 192,
3337
+ 1
3338
+ ],
3339
+ "storage": "direct",
3340
+ "target": "dp.convs.convs_1x1.2.weight",
3341
+ "transform": "identity"
3342
+ },
3343
+ {
3344
+ "initializer": "dp.convs.convs_1x1.2.bias",
3345
+ "shape": [
3346
+ 192
3347
+ ],
3348
+ "storage": "direct",
3349
+ "target": "dp.convs.convs_1x1.2.bias",
3350
+ "transform": "identity"
3351
+ },
3352
+ {
3353
+ "initializer": "dp.convs.norms_1.0.gamma",
3354
+ "shape": [
3355
+ 192
3356
+ ],
3357
+ "storage": "direct",
3358
+ "target": "dp.convs.norms_1.0.gamma",
3359
+ "transform": "identity"
3360
+ },
3361
+ {
3362
+ "initializer": "dp.convs.norms_1.0.beta",
3363
+ "shape": [
3364
+ 192
3365
+ ],
3366
+ "storage": "direct",
3367
+ "target": "dp.convs.norms_1.0.beta",
3368
+ "transform": "identity"
3369
+ },
3370
+ {
3371
+ "initializer": "dp.convs.norms_1.1.gamma",
3372
+ "shape": [
3373
+ 192
3374
+ ],
3375
+ "storage": "direct",
3376
+ "target": "dp.convs.norms_1.1.gamma",
3377
+ "transform": "identity"
3378
+ },
3379
+ {
3380
+ "initializer": "dp.convs.norms_1.1.beta",
3381
+ "shape": [
3382
+ 192
3383
+ ],
3384
+ "storage": "direct",
3385
+ "target": "dp.convs.norms_1.1.beta",
3386
+ "transform": "identity"
3387
+ },
3388
+ {
3389
+ "initializer": "dp.convs.norms_1.2.gamma",
3390
+ "shape": [
3391
+ 192
3392
+ ],
3393
+ "storage": "direct",
3394
+ "target": "dp.convs.norms_1.2.gamma",
3395
+ "transform": "identity"
3396
+ },
3397
+ {
3398
+ "initializer": "dp.convs.norms_1.2.beta",
3399
+ "shape": [
3400
+ 192
3401
+ ],
3402
+ "storage": "direct",
3403
+ "target": "dp.convs.norms_1.2.beta",
3404
+ "transform": "identity"
3405
+ },
3406
+ {
3407
+ "initializer": "dp.convs.norms_2.0.gamma",
3408
+ "shape": [
3409
+ 192
3410
+ ],
3411
+ "storage": "direct",
3412
+ "target": "dp.convs.norms_2.0.gamma",
3413
+ "transform": "identity"
3414
+ },
3415
+ {
3416
+ "initializer": "dp.convs.norms_2.0.beta",
3417
+ "shape": [
3418
+ 192
3419
+ ],
3420
+ "storage": "direct",
3421
+ "target": "dp.convs.norms_2.0.beta",
3422
+ "transform": "identity"
3423
+ },
3424
+ {
3425
+ "initializer": "dp.convs.norms_2.1.gamma",
3426
+ "shape": [
3427
+ 192
3428
+ ],
3429
+ "storage": "direct",
3430
+ "target": "dp.convs.norms_2.1.gamma",
3431
+ "transform": "identity"
3432
+ },
3433
+ {
3434
+ "initializer": "dp.convs.norms_2.1.beta",
3435
+ "shape": [
3436
+ 192
3437
+ ],
3438
+ "storage": "direct",
3439
+ "target": "dp.convs.norms_2.1.beta",
3440
+ "transform": "identity"
3441
+ },
3442
+ {
3443
+ "initializer": "dp.convs.norms_2.2.gamma",
3444
+ "shape": [
3445
+ 192
3446
+ ],
3447
+ "storage": "direct",
3448
+ "target": "dp.convs.norms_2.2.gamma",
3449
+ "transform": "identity"
3450
+ },
3451
+ {
3452
+ "initializer": "dp.convs.norms_2.2.beta",
3453
+ "shape": [
3454
+ 192
3455
+ ],
3456
+ "storage": "direct",
3457
+ "target": "dp.convs.norms_2.2.beta",
3458
+ "transform": "identity"
3459
+ },
3460
+ {
3461
+ "initializer": "onnx::Exp_8793",
3462
+ "shape": [
3463
+ 2,
3464
+ 1
3465
+ ],
3466
+ "storage": "direct",
3467
+ "target": "dp.flows.0.logs",
3468
+ "transform": "negate"
3469
+ }
3470
+ ],
3471
+ "training_only_initialization": {
3472
+ "modules": [
3473
+ "model_g.enc_q",
3474
+ "model_g.dp training-only branches",
3475
+ "model_d"
3476
+ ],
3477
+ "source": "Mana parent checkpoint"
3478
+ }
3479
+ }
training/train_rizeh_full_100k.sh ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ set -euo pipefail
3
+
4
+ root=/home/rezo/piper-wikipedia-approved-20260830
5
+ data="${root}/data/aligned-full-v1"
6
+ signal_run="${root}/runs/rizeh-recovered-approved-signal-2000"
7
+ run="${root}/runs/rizeh-recovered-approved-full-100k"
8
+ resume_checkpoint="${signal_run}/lightning_logs/version_1/checkpoints/last.ckpt"
9
+
10
+ [[ -s "${resume_checkpoint}" ]]
11
+ [[ "$(jq -r .status "${data}/report.json")" == complete ]]
12
+ mkdir -p "${run}"
13
+
14
+ cd "${root}/src/piper1-gpl"
15
+ "${root}/venv/bin/python" -m piper.train fit \
16
+ --trainer.accelerator gpu \
17
+ --trainer.devices 1 \
18
+ --trainer.precision bf16-mixed \
19
+ --trainer.max_steps 100000 \
20
+ --trainer.num_sanity_val_steps 0 \
21
+ --trainer.val_check_interval 500 \
22
+ --trainer.limit_val_batches 25 \
23
+ --trainer.log_every_n_steps 10 \
24
+ --trainer.enable_progress_bar false \
25
+ --trainer.default_root_dir "${run}" \
26
+ --model.sample_rate 22050 \
27
+ --model.num_speakers 1 \
28
+ --model.warmstart_ckpt null \
29
+ --model.learning_rate 0.00001 \
30
+ --model.learning_rate_d 0.000005 \
31
+ --model.mos_metric null \
32
+ --data.voice_name gooya-rizehpizeh-wikipedia-kore-full-100k \
33
+ --data.csv_path "${data}/metadata-train.csv" \
34
+ --data.audio_dir / \
35
+ --data.cache_dir "${root}/cache/aligned-full-v1" \
36
+ --data.config_path "${signal_run}/model.onnx.json" \
37
+ --data.batch_size 1 \
38
+ --data.validation_split 0.01 \
39
+ --data.num_test_examples 0 \
40
+ --data.num_workers 2 \
41
+ --data.trim_silence false \
42
+ --data.phoneme_type text \
43
+ --data.dataset_type phoneme_ids \
44
+ --data.phonemes_path "${data}/negara-phoneme-id-map.json" \
45
+ --data.espeak_voice fa \
46
+ --data.num_symbols 256 \
47
+ --ckpt_path "${resume_checkpoint}"
48
+
49
+ last_checkpoint=$(find "${run}" -name last.ckpt -type f -printf '%T@ %p\n' | sort -nr | head -1 | cut -d' ' -f2-)
50
+ best_checkpoint=$("${root}/venv/bin/python" - "${run}" <<'PY'
51
+ import re
52
+ import sys
53
+ from pathlib import Path
54
+
55
+ run = Path(sys.argv[1])
56
+ candidates = []
57
+ for path in run.glob("lightning_logs/version_*/checkpoints/*val_mel=*.ckpt"):
58
+ match = re.search(r"val_mel=([0-9.]+)\.ckpt$", path.name)
59
+ if match:
60
+ candidates.append((float(match.group(1)), str(path)))
61
+ if not candidates:
62
+ raise SystemExit("No validation checkpoint found")
63
+ print(min(candidates)[1])
64
+ PY
65
+ )
66
+ [[ -s "${last_checkpoint}" ]]
67
+ [[ -s "${best_checkpoint}" ]]
68
+
69
+ "${root}/venv/bin/python" -m piper.train.export_onnx \
70
+ --checkpoint "${best_checkpoint}" \
71
+ --output-file "${run}/model-best.onnx"
72
+
73
+ printf '%s\n' 'zurxAneye vahdat yek makAne varzeSi dar Sahre borujerd ast.' | \
74
+ "${root}/venv/bin/python" -m piper \
75
+ -m "${run}/model-best.onnx" \
76
+ -f "${run}/zurxane-full-100k-best.wav" \
77
+ --length-scale 1.15
78
+
79
+ sha256sum "${last_checkpoint}" "${best_checkpoint}" "${run}/model-best.onnx" \
80
+ "${run}/model-best.onnx.json" "${run}/zurxane-full-100k-best.wav" \
81
+ > "${run}/sha256sums.txt"
82
+ echo "full_complete_utc=$(date -u +%FT%TZ) last=${last_checkpoint} best=${best_checkpoint}"