| { |
| "inputs": { |
| "mapping_reference_checkpoint": "/home/rezo/piper-wikipedia-approved-20260830/runs/mana-warmstart-pilot3-runtimecheck/lightning_logs/version_0/checkpoints/last.ckpt", |
| "mapping_reference_checkpoint_sha256": "720e6f1deb9b665fa3a64b849578e67b22e5cd5457589492748c59b5af629239", |
| "mapping_reference_onnx": "/home/rezo/piper-wikipedia-approved-20260830/runs/mana-warmstart-pilot3-runtimecheck/model.onnx", |
| "mapping_reference_onnx_sha256": "b23dde224885d42fdddec8aa886e24fc8fa9cde7f188622e2771eb0ff2d4c2be", |
| "production_onnx": "/home/rezo/piper-wikipedia-approved-20260830/models/rizeh-onnx-recovery/epoch4-rizehpizeh.onnx", |
| "production_onnx_sha256": "dc8532f0c57a40ed076e466f380376577e8bbcd13d1fc19c81f69d461d26dadc", |
| "template_checkpoint": "/home/rezo/piper-wikipedia-approved-20260830/models/mana/ckpt/epoch=6012-step=4203520.ckpt", |
| "template_checkpoint_sha256": "ec2c289b8f4a32b0a7f1f491a98beaa24f8fab8a13de537495f1d649c8818980" |
| }, |
| "output_checkpoint": "/home/rezo/piper-wikipedia-approved-20260830/models/rizeh-onnx-recovery/rizehpizeh-epoch4-recovered-trainable.ckpt", |
| "output_checkpoint_sha256": "331c1e8c95e86f88f2a11a6eb14475da609a58891a014cf3cf1e3af414210347", |
| "recovered_by_role": { |
| "dec": 45, |
| "dp": 114, |
| "enc_p": 111, |
| "flow": 80 |
| }, |
| "recovered_inference_tensors": 350, |
| "schema_version": "gooya.rizeh-onnx-recovery/v1", |
| "special_transform_verification": { |
| "initializer": "onnx::Exp_8946", |
| "max_abs_error": 0.0, |
| "transform": "checkpoint_value = -onnx_initializer" |
| }, |
| "status": "recovered_pending_roundtrip_parity", |
| "tensors": [ |
| { |
| "initializer": "sid", |
| "shape": [ |
| 256, |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.emb.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.0.emb_rel_k", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.0.emb_rel_k", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.0.emb_rel_v", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.0.emb_rel_v", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.0.conv_q.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.0.conv_q.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.0.conv_q.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.0.conv_q.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.0.conv_k.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.0.conv_k.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.0.conv_k.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.0.conv_k.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.0.conv_v.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.0.conv_v.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.0.conv_v.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.0.conv_v.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.0.conv_o.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.0.conv_o.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.0.conv_o.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.0.conv_o.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.1.emb_rel_k", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.1.emb_rel_k", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.1.emb_rel_v", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.1.emb_rel_v", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.1.conv_q.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.1.conv_q.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.1.conv_q.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.1.conv_q.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.1.conv_k.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.1.conv_k.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.1.conv_k.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.1.conv_k.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.1.conv_v.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.1.conv_v.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.1.conv_v.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.1.conv_v.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.1.conv_o.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.1.conv_o.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.1.conv_o.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.1.conv_o.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.2.emb_rel_k", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.2.emb_rel_k", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.2.emb_rel_v", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.2.emb_rel_v", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.2.conv_q.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.2.conv_q.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.2.conv_q.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.2.conv_q.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.2.conv_k.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.2.conv_k.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.2.conv_k.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.2.conv_k.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.2.conv_v.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.2.conv_v.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.2.conv_v.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.2.conv_v.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.2.conv_o.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.2.conv_o.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.2.conv_o.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.2.conv_o.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.3.emb_rel_k", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.3.emb_rel_k", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.3.emb_rel_v", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.3.emb_rel_v", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.3.conv_q.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.3.conv_q.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.3.conv_q.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.3.conv_q.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.3.conv_k.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.3.conv_k.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.3.conv_k.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.3.conv_k.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.3.conv_v.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.3.conv_v.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.3.conv_v.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.3.conv_v.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.3.conv_o.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.3.conv_o.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.3.conv_o.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.3.conv_o.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.4.emb_rel_k", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.4.emb_rel_k", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.4.emb_rel_v", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.4.emb_rel_v", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.4.conv_q.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.4.conv_q.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.4.conv_q.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.4.conv_q.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.4.conv_k.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.4.conv_k.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.4.conv_k.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.4.conv_k.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.4.conv_v.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.4.conv_v.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.4.conv_v.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.4.conv_v.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.4.conv_o.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.4.conv_o.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.4.conv_o.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.4.conv_o.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.5.emb_rel_k", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.5.emb_rel_k", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.5.emb_rel_v", |
| "shape": [ |
| 1, |
| 9, |
| 96 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.5.emb_rel_v", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.5.conv_q.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.5.conv_q.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.5.conv_q.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.5.conv_q.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.5.conv_k.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.5.conv_k.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.5.conv_k.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.5.conv_k.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.5.conv_v.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.5.conv_v.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.5.conv_v.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.5.conv_v.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.5.conv_o.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.5.conv_o.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.attn_layers.5.conv_o.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.attn_layers.5.conv_o.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.0.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.0.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.0.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.0.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.1.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.1.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.1.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.1.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.2.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.2.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.2.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.2.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.3.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.3.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.3.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.3.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.4.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.4.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.4.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.4.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.5.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.5.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_1.5.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_1.5.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.0.conv_1.weight", |
| "shape": [ |
| 768, |
| 192, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.0.conv_1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.0.conv_1.bias", |
| "shape": [ |
| 768 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.0.conv_1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.0.conv_2.weight", |
| "shape": [ |
| 192, |
| 768, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.0.conv_2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.0.conv_2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.0.conv_2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.1.conv_1.weight", |
| "shape": [ |
| 768, |
| 192, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.1.conv_1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.1.conv_1.bias", |
| "shape": [ |
| 768 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.1.conv_1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.1.conv_2.weight", |
| "shape": [ |
| 192, |
| 768, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.1.conv_2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.1.conv_2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.1.conv_2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.2.conv_1.weight", |
| "shape": [ |
| 768, |
| 192, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.2.conv_1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.2.conv_1.bias", |
| "shape": [ |
| 768 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.2.conv_1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.2.conv_2.weight", |
| "shape": [ |
| 192, |
| 768, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.2.conv_2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.2.conv_2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.2.conv_2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.3.conv_1.weight", |
| "shape": [ |
| 768, |
| 192, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.3.conv_1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.3.conv_1.bias", |
| "shape": [ |
| 768 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.3.conv_1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.3.conv_2.weight", |
| "shape": [ |
| 192, |
| 768, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.3.conv_2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.3.conv_2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.3.conv_2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.4.conv_1.weight", |
| "shape": [ |
| 768, |
| 192, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.4.conv_1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.4.conv_1.bias", |
| "shape": [ |
| 768 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.4.conv_1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.4.conv_2.weight", |
| "shape": [ |
| 192, |
| 768, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.4.conv_2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.4.conv_2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.4.conv_2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.5.conv_1.weight", |
| "shape": [ |
| 768, |
| 192, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.5.conv_1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.5.conv_1.bias", |
| "shape": [ |
| 768 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.5.conv_1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.5.conv_2.weight", |
| "shape": [ |
| 192, |
| 768, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.5.conv_2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.ffn_layers.5.conv_2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.ffn_layers.5.conv_2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.0.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.0.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.0.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.0.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.1.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.1.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.1.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.1.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.2.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.2.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.2.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.2.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.3.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.3.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.3.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.3.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.4.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.4.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.4.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.4.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.5.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.5.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.encoder.norm_layers_2.5.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "enc_p.encoder.norm_layers_2.5.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.proj.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "enc_p.proj.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "enc_p.proj.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "enc_p.proj.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.conv_pre.weight", |
| "shape": [ |
| 256, |
| 192, |
| 7 |
| ], |
| "storage": "direct", |
| "target": "dec.conv_pre.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.conv_pre.bias", |
| "shape": [ |
| 256 |
| ], |
| "storage": "direct", |
| "target": "dec.conv_pre.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.ups.0.bias", |
| "shape": [ |
| 128 |
| ], |
| "storage": "direct", |
| "target": "dec.ups.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.ups.0.weight", |
| "shape": [ |
| 256, |
| 128, |
| 16 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.ups.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.ups.1.bias", |
| "shape": [ |
| 64 |
| ], |
| "storage": "direct", |
| "target": "dec.ups.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.ups.1.weight", |
| "shape": [ |
| 128, |
| 64, |
| 16 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.ups.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.ups.2.bias", |
| "shape": [ |
| 32 |
| ], |
| "storage": "direct", |
| "target": "dec.ups.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.ups.2.weight", |
| "shape": [ |
| 64, |
| 32, |
| 8 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.ups.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.0.convs.0.bias", |
| "shape": [ |
| 128 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.0.convs.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.0.convs.0.weight", |
| "shape": [ |
| 128, |
| 128, |
| 3 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.0.convs.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.0.convs.1.bias", |
| "shape": [ |
| 128 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.0.convs.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.0.convs.1.weight", |
| "shape": [ |
| 128, |
| 128, |
| 3 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.0.convs.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.1.convs.0.bias", |
| "shape": [ |
| 128 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.1.convs.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.1.convs.0.weight", |
| "shape": [ |
| 128, |
| 128, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.1.convs.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.1.convs.1.bias", |
| "shape": [ |
| 128 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.1.convs.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.1.convs.1.weight", |
| "shape": [ |
| 128, |
| 128, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.1.convs.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.2.convs.0.bias", |
| "shape": [ |
| 128 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.2.convs.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.2.convs.0.weight", |
| "shape": [ |
| 128, |
| 128, |
| 7 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.2.convs.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.2.convs.1.bias", |
| "shape": [ |
| 128 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.2.convs.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.2.convs.1.weight", |
| "shape": [ |
| 128, |
| 128, |
| 7 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.2.convs.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.3.convs.0.bias", |
| "shape": [ |
| 64 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.3.convs.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.3.convs.0.weight", |
| "shape": [ |
| 64, |
| 64, |
| 3 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.3.convs.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.3.convs.1.bias", |
| "shape": [ |
| 64 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.3.convs.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.3.convs.1.weight", |
| "shape": [ |
| 64, |
| 64, |
| 3 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.3.convs.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.4.convs.0.bias", |
| "shape": [ |
| 64 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.4.convs.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.4.convs.0.weight", |
| "shape": [ |
| 64, |
| 64, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.4.convs.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.4.convs.1.bias", |
| "shape": [ |
| 64 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.4.convs.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.4.convs.1.weight", |
| "shape": [ |
| 64, |
| 64, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.4.convs.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.5.convs.0.bias", |
| "shape": [ |
| 64 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.5.convs.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.5.convs.0.weight", |
| "shape": [ |
| 64, |
| 64, |
| 7 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.5.convs.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.5.convs.1.bias", |
| "shape": [ |
| 64 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.5.convs.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.5.convs.1.weight", |
| "shape": [ |
| 64, |
| 64, |
| 7 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.5.convs.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.6.convs.0.bias", |
| "shape": [ |
| 32 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.6.convs.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.6.convs.0.weight", |
| "shape": [ |
| 32, |
| 32, |
| 3 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.6.convs.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.6.convs.1.bias", |
| "shape": [ |
| 32 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.6.convs.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.6.convs.1.weight", |
| "shape": [ |
| 32, |
| 32, |
| 3 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.6.convs.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.7.convs.0.bias", |
| "shape": [ |
| 32 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.7.convs.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.7.convs.0.weight", |
| "shape": [ |
| 32, |
| 32, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.7.convs.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.7.convs.1.bias", |
| "shape": [ |
| 32 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.7.convs.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.7.convs.1.weight", |
| "shape": [ |
| 32, |
| 32, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.7.convs.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.8.convs.0.bias", |
| "shape": [ |
| 32 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.8.convs.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.8.convs.0.weight", |
| "shape": [ |
| 32, |
| 32, |
| 7 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.8.convs.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.8.convs.1.bias", |
| "shape": [ |
| 32 |
| ], |
| "storage": "direct", |
| "target": "dec.resblocks.8.convs.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.resblocks.8.convs.1.weight", |
| "shape": [ |
| 32, |
| 32, |
| 7 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "dec.resblocks.8.convs.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dec.conv_post.weight", |
| "shape": [ |
| 1, |
| 32, |
| 7 |
| ], |
| "storage": "direct", |
| "target": "dec.conv_post.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.pre.weight", |
| "shape": [ |
| 192, |
| 96, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.pre.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.pre.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.pre.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.in_layers.0.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.enc.in_layers.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.in_layers.0.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.0.enc.in_layers.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.in_layers.1.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.enc.in_layers.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.in_layers.1.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.0.enc.in_layers.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.in_layers.2.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.enc.in_layers.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.in_layers.2.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.0.enc.in_layers.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.in_layers.3.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.enc.in_layers.3.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.in_layers.3.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.0.enc.in_layers.3.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.res_skip_layers.0.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.enc.res_skip_layers.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.res_skip_layers.0.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.0.enc.res_skip_layers.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.res_skip_layers.1.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.enc.res_skip_layers.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.res_skip_layers.1.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.0.enc.res_skip_layers.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.res_skip_layers.2.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.enc.res_skip_layers.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.res_skip_layers.2.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.0.enc.res_skip_layers.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.res_skip_layers.3.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.enc.res_skip_layers.3.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.enc.res_skip_layers.3.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.0.enc.res_skip_layers.3.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.post.weight", |
| "shape": [ |
| 96, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.post.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.0.post.bias", |
| "shape": [ |
| 96 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.0.post.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.pre.weight", |
| "shape": [ |
| 192, |
| 96, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.pre.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.pre.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.pre.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.in_layers.0.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.enc.in_layers.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.in_layers.0.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.2.enc.in_layers.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.in_layers.1.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.enc.in_layers.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.in_layers.1.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.2.enc.in_layers.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.in_layers.2.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.enc.in_layers.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.in_layers.2.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.2.enc.in_layers.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.in_layers.3.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.enc.in_layers.3.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.in_layers.3.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.2.enc.in_layers.3.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.res_skip_layers.0.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.enc.res_skip_layers.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.res_skip_layers.0.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.2.enc.res_skip_layers.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.res_skip_layers.1.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.enc.res_skip_layers.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.res_skip_layers.1.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.2.enc.res_skip_layers.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.res_skip_layers.2.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.enc.res_skip_layers.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.res_skip_layers.2.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.2.enc.res_skip_layers.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.res_skip_layers.3.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.enc.res_skip_layers.3.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.enc.res_skip_layers.3.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.2.enc.res_skip_layers.3.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.post.weight", |
| "shape": [ |
| 96, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.post.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.2.post.bias", |
| "shape": [ |
| 96 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.2.post.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.pre.weight", |
| "shape": [ |
| 192, |
| 96, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.pre.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.pre.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.pre.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.in_layers.0.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.enc.in_layers.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.in_layers.0.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.4.enc.in_layers.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.in_layers.1.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.enc.in_layers.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.in_layers.1.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.4.enc.in_layers.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.in_layers.2.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.enc.in_layers.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.in_layers.2.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.4.enc.in_layers.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.in_layers.3.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.enc.in_layers.3.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.in_layers.3.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.4.enc.in_layers.3.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.res_skip_layers.0.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.enc.res_skip_layers.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.res_skip_layers.0.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.4.enc.res_skip_layers.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.res_skip_layers.1.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.enc.res_skip_layers.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.res_skip_layers.1.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.4.enc.res_skip_layers.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.res_skip_layers.2.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.enc.res_skip_layers.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.res_skip_layers.2.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.4.enc.res_skip_layers.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.res_skip_layers.3.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.enc.res_skip_layers.3.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.enc.res_skip_layers.3.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.4.enc.res_skip_layers.3.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.post.weight", |
| "shape": [ |
| 96, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.post.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.4.post.bias", |
| "shape": [ |
| 96 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.4.post.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.pre.weight", |
| "shape": [ |
| 192, |
| 96, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.pre.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.pre.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.pre.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.in_layers.0.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.enc.in_layers.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.in_layers.0.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.6.enc.in_layers.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.in_layers.1.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.enc.in_layers.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.in_layers.1.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.6.enc.in_layers.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.in_layers.2.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.enc.in_layers.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.in_layers.2.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.6.enc.in_layers.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.in_layers.3.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.enc.in_layers.3.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.in_layers.3.weight", |
| "shape": [ |
| 384, |
| 192, |
| 5 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.6.enc.in_layers.3.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.res_skip_layers.0.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.enc.res_skip_layers.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.res_skip_layers.0.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.6.enc.res_skip_layers.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.res_skip_layers.1.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.enc.res_skip_layers.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.res_skip_layers.1.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.6.enc.res_skip_layers.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.res_skip_layers.2.bias", |
| "shape": [ |
| 384 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.enc.res_skip_layers.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.res_skip_layers.2.weight", |
| "shape": [ |
| 384, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.6.enc.res_skip_layers.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.res_skip_layers.3.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.enc.res_skip_layers.3.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.enc.res_skip_layers.3.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "weight_norm_reconstructed", |
| "target": "flow.flows.6.enc.res_skip_layers.3.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.post.weight", |
| "shape": [ |
| 96, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.post.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "flow.flows.6.post.bias", |
| "shape": [ |
| 96 |
| ], |
| "storage": "direct", |
| "target": "flow.flows.6.post.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.0.m", |
| "shape": [ |
| 2, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.0.m", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.pre.weight", |
| "shape": [ |
| 192, |
| 1, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.pre.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.pre.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.pre.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_sep.0.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_sep.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_sep.0.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_sep.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_sep.1.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_sep.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_sep.1.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_sep.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_sep.2.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_sep.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_sep.2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_sep.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_1x1.0.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_1x1.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_1x1.0.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_1x1.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_1x1.1.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_1x1.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_1x1.1.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_1x1.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_1x1.2.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_1x1.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.convs_1x1.2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.convs_1x1.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_1.0.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_1.0.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_1.0.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_1.0.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_1.1.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_1.1.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_1.1.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_1.1.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_1.2.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_1.2.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_1.2.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_1.2.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_2.0.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_2.0.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_2.0.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_2.0.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_2.1.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_2.1.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_2.1.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_2.1.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_2.2.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_2.2.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.convs.norms_2.2.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.convs.norms_2.2.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.proj.weight", |
| "shape": [ |
| 29, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.proj.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.3.proj.bias", |
| "shape": [ |
| 29 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.3.proj.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.pre.weight", |
| "shape": [ |
| 192, |
| 1, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.pre.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.pre.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.pre.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_sep.0.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_sep.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_sep.0.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_sep.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_sep.1.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_sep.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_sep.1.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_sep.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_sep.2.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_sep.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_sep.2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_sep.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_1x1.0.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_1x1.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_1x1.0.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_1x1.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_1x1.1.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_1x1.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_1x1.1.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_1x1.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_1x1.2.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_1x1.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.convs_1x1.2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.convs_1x1.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_1.0.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_1.0.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_1.0.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_1.0.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_1.1.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_1.1.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_1.1.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_1.1.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_1.2.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_1.2.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_1.2.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_1.2.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_2.0.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_2.0.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_2.0.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_2.0.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_2.1.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_2.1.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_2.1.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_2.1.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_2.2.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_2.2.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.convs.norms_2.2.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.convs.norms_2.2.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.proj.weight", |
| "shape": [ |
| 29, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.proj.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.5.proj.bias", |
| "shape": [ |
| 29 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.5.proj.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.pre.weight", |
| "shape": [ |
| 192, |
| 1, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.pre.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.pre.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.pre.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_sep.0.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_sep.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_sep.0.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_sep.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_sep.1.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_sep.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_sep.1.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_sep.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_sep.2.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_sep.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_sep.2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_sep.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_1x1.0.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_1x1.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_1x1.0.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_1x1.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_1x1.1.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_1x1.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_1x1.1.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_1x1.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_1x1.2.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_1x1.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.convs_1x1.2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.convs_1x1.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_1.0.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_1.0.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_1.0.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_1.0.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_1.1.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_1.1.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_1.1.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_1.1.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_1.2.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_1.2.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_1.2.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_1.2.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_2.0.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_2.0.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_2.0.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_2.0.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_2.1.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_2.1.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_2.1.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_2.1.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_2.2.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_2.2.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.convs.norms_2.2.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.convs.norms_2.2.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.proj.weight", |
| "shape": [ |
| 29, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.proj.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.flows.7.proj.bias", |
| "shape": [ |
| 29 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.7.proj.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.pre.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.pre.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.pre.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.pre.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.proj.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.proj.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.proj.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.proj.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_sep.0.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_sep.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_sep.0.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_sep.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_sep.1.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_sep.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_sep.1.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_sep.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_sep.2.weight", |
| "shape": [ |
| 192, |
| 1, |
| 3 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_sep.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_sep.2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_sep.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_1x1.0.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_1x1.0.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_1x1.0.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_1x1.0.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_1x1.1.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_1x1.1.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_1x1.1.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_1x1.1.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_1x1.2.weight", |
| "shape": [ |
| 192, |
| 192, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_1x1.2.weight", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.convs_1x1.2.bias", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.convs_1x1.2.bias", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_1.0.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_1.0.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_1.0.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_1.0.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_1.1.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_1.1.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_1.1.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_1.1.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_1.2.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_1.2.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_1.2.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_1.2.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_2.0.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_2.0.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_2.0.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_2.0.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_2.1.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_2.1.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_2.1.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_2.1.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_2.2.gamma", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_2.2.gamma", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "dp.convs.norms_2.2.beta", |
| "shape": [ |
| 192 |
| ], |
| "storage": "direct", |
| "target": "dp.convs.norms_2.2.beta", |
| "transform": "identity" |
| }, |
| { |
| "initializer": "onnx::Exp_8793", |
| "shape": [ |
| 2, |
| 1 |
| ], |
| "storage": "direct", |
| "target": "dp.flows.0.logs", |
| "transform": "negate" |
| } |
| ], |
| "training_only_initialization": { |
| "modules": [ |
| "model_g.enc_q", |
| "model_g.dp training-only branches", |
| "model_d" |
| ], |
| "source": "Mana parent checkpoint" |
| } |
| } |
|
|
