vae.pt/array_metadatas/process_11

{"array_metadatas": [{"array_metadata": {"param_name": "conv1.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "conv1.kernel.value", "write_shape": [1, 1, 1, 32, 32], "chunk_shape": [1, 1, 1, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "conv2.bias.value", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "conv2.kernel.value", "write_shape": [1, 1, 1, 16, 16], "chunk_shape": [1, 1, 1, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.conv1.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.conv1.kernel.value", "write_shape": [3, 3, 3, 16, 6], "chunk_shape": [3, 3, 3, 16, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.head.0.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.head.2.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.head.2.kernel.value", "write_shape": [3, 3, 3, 96, 3], "chunk_shape": [3, 3, 3, 96, 3], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.0.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.0.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.0.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.0.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.0.residual.5.rngs.default.count.value", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.0.residual.5.rngs.default.key.value", "write_shape": [2], "chunk_shape": [2], "ext_metadata": {"random_key_impl": "threefry2x32"}}}, {"array_metadata": {"param_name": "decoder.middle.0.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.0.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.1.norm.gamma.value", "write_shape": [6, 1, 1], "chunk_shape": [6, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.1.proj.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.1.proj.kernel.value", "write_shape": [1, 1, 6, 384], "chunk_shape": [1, 1, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.1.to_qkv.bias.value", "write_shape": [18], "chunk_shape": [18], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.1.to_qkv.kernel.value", "write_shape": [1, 1, 6, 1152], "chunk_shape": [1, 1, 6, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.2.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.2.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.2.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.2.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.2.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.middle.2.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.0.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.0.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.0.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.0.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.0.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.0.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.1.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.1.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.1.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.1.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.1.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.1.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.2.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.2.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.2.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.2.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.2.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.2.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.3.conv.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.3.conv.kernel.value", "write_shape": [3, 3, 6, 192], "chunk_shape": [3, 3, 6, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.3.time_conv.bias.value", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.3.time_conv.kernel.value", "write_shape": [3, 1, 1, 6, 768], "chunk_shape": [3, 1, 1, 6, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.4.residual.0.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.4.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.4.residual.2.kernel.value", "write_shape": [3, 3, 3, 3, 384], "chunk_shape": [3, 3, 3, 3, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.4.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.4.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.4.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.4.shortcut.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.4.shortcut.kernel.value", "write_shape": [1, 1, 1, 3, 384], "chunk_shape": [1, 1, 1, 3, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.5.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.5.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.5.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.5.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.5.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.5.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.6.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.6.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.6.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.6.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.6.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.6.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.7.conv.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.7.conv.kernel.value", "write_shape": [3, 3, 6, 192], "chunk_shape": [3, 3, 6, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.7.time_conv.bias.value", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.7.time_conv.kernel.value", "write_shape": [3, 1, 1, 6, 768], "chunk_shape": [3, 1, 1, 6, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.8.residual.0.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.8.residual.2.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.8.residual.2.kernel.value", "write_shape": [3, 3, 3, 3, 192], "chunk_shape": [3, 3, 3, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.8.residual.3.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.8.residual.6.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.8.residual.6.kernel.value", "write_shape": [3, 3, 3, 3, 192], "chunk_shape": [3, 3, 3, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.9.residual.0.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.9.residual.2.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.9.residual.2.kernel.value", "write_shape": [3, 3, 3, 3, 192], "chunk_shape": [3, 3, 3, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.9.residual.3.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.9.residual.6.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.9.residual.6.kernel.value", "write_shape": [3, 3, 3, 3, 192], "chunk_shape": [3, 3, 3, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.10.residual.0.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.10.residual.2.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.10.residual.2.kernel.value", "write_shape": [3, 3, 3, 3, 192], "chunk_shape": [3, 3, 3, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.10.residual.3.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.10.residual.6.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.10.residual.6.kernel.value", "write_shape": [3, 3, 3, 3, 192], "chunk_shape": [3, 3, 3, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.11.conv.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.11.conv.kernel.value", "write_shape": [3, 3, 3, 96], "chunk_shape": [3, 3, 3, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.12.residual.0.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.12.residual.2.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.12.residual.2.kernel.value", "write_shape": [3, 3, 3, 96, 96], "chunk_shape": [3, 3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.12.residual.3.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.12.residual.6.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.12.residual.6.kernel.value", "write_shape": [3, 3, 3, 96, 96], "chunk_shape": [3, 3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.13.residual.0.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.13.residual.2.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.13.residual.2.kernel.value", "write_shape": [3, 3, 3, 96, 96], "chunk_shape": [3, 3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.13.residual.3.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.13.residual.6.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.13.residual.6.kernel.value", "write_shape": [3, 3, 3, 96, 96], "chunk_shape": [3, 3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.14.residual.0.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.14.residual.2.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.14.residual.2.kernel.value", "write_shape": [3, 3, 3, 96, 96], "chunk_shape": [3, 3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.14.residual.3.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.14.residual.6.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "decoder.upsamples.14.residual.6.kernel.value", "write_shape": [3, 3, 3, 96, 96], "chunk_shape": [3, 3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.conv1.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.conv1.kernel.value", "write_shape": [3, 3, 3, 3, 96], "chunk_shape": [3, 3, 3, 3, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.0.residual.0.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.0.residual.2.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.0.residual.2.kernel.value", "write_shape": [3, 3, 3, 96, 96], "chunk_shape": [3, 3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.0.residual.3.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.0.residual.6.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.0.residual.6.kernel.value", "write_shape": [3, 3, 3, 96, 96], "chunk_shape": [3, 3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.1.residual.0.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.1.residual.2.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.1.residual.2.kernel.value", "write_shape": [3, 3, 3, 96, 96], "chunk_shape": [3, 3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.1.residual.3.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.1.residual.6.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.1.residual.6.kernel.value", "write_shape": [3, 3, 3, 96, 96], "chunk_shape": [3, 3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.2.conv.bias.value", "write_shape": [96], "chunk_shape": [96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.2.conv.kernel.value", "write_shape": [3, 3, 96, 96], "chunk_shape": [3, 3, 96, 96], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.3.residual.0.gamma.value", "write_shape": [96, 1, 1, 1], "chunk_shape": [96, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.3.residual.2.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.3.residual.2.kernel.value", "write_shape": [3, 3, 3, 96, 3], "chunk_shape": [3, 3, 3, 96, 3], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.3.residual.3.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.3.residual.6.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.3.residual.6.kernel.value", "write_shape": [3, 3, 3, 3, 192], "chunk_shape": [3, 3, 3, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.3.shortcut.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.3.shortcut.kernel.value", "write_shape": [1, 1, 1, 96, 3], "chunk_shape": [1, 1, 1, 96, 3], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.4.residual.0.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.4.residual.2.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.4.residual.2.kernel.value", "write_shape": [3, 3, 3, 3, 192], "chunk_shape": [3, 3, 3, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.4.residual.3.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.4.residual.6.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.4.residual.6.kernel.value", "write_shape": [3, 3, 3, 3, 192], "chunk_shape": [3, 3, 3, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.5.conv.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.5.conv.kernel.value", "write_shape": [3, 3, 3, 192], "chunk_shape": [3, 3, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.5.time_conv.bias.value", "write_shape": [3], "chunk_shape": [3], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.5.time_conv.kernel.value", "write_shape": [3, 1, 1, 3, 192], "chunk_shape": [3, 1, 1, 3, 192], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.6.residual.0.gamma.value", "write_shape": [3, 1, 1, 1], "chunk_shape": [3, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.6.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.6.residual.2.kernel.value", "write_shape": [3, 3, 3, 3, 384], "chunk_shape": [3, 3, 3, 3, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.6.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.6.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.6.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.6.shortcut.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.6.shortcut.kernel.value", "write_shape": [1, 1, 1, 3, 384], "chunk_shape": [1, 1, 1, 3, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.7.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.7.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.7.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.7.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.7.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.7.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.8.conv.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.8.conv.kernel.value", "write_shape": [3, 3, 6, 384], "chunk_shape": [3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.8.time_conv.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.8.time_conv.kernel.value", "write_shape": [3, 1, 1, 6, 384], "chunk_shape": [3, 1, 1, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.9.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.9.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.9.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.9.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.9.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.9.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.10.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.10.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.10.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.10.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.10.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.downsamples.10.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.head.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.head.2.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.head.2.kernel.value", "write_shape": [3, 3, 3, 6, 32], "chunk_shape": [3, 3, 3, 6, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.0.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.0.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.0.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.0.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.0.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.0.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.1.norm.gamma.value", "write_shape": [6, 1, 1], "chunk_shape": [6, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.1.proj.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.1.proj.kernel.value", "write_shape": [1, 1, 6, 384], "chunk_shape": [1, 1, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.1.to_qkv.bias.value", "write_shape": [18], "chunk_shape": [18], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.1.to_qkv.kernel.value", "write_shape": [1, 1, 6, 1152], "chunk_shape": [1, 1, 6, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.2.residual.0.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.2.residual.2.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.2.residual.2.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.2.residual.3.gamma.value", "write_shape": [6, 1, 1, 1], "chunk_shape": [6, 1, 1, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.2.residual.6.bias.value", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.middle.2.residual.6.kernel.value", "write_shape": [3, 3, 3, 6, 384], "chunk_shape": [3, 3, 3, 6, 384], "ext_metadata": null}}]}