Upload model weights with LFS

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+**/d/* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +1,39 @@
 ---
+language:
+- en
 license: mit
+library_name: jax
+tags:
+- robotics
+- reinforcement-learning
+- continuous-control
+- test-time-adaptation
+- flow-matching
+- orbax
 ---
+# Model Card for quadruped_domain_randomization_model
+
+## Model Description
+This repository contains the pre-trained neural network weights for the **Reversible Flow Adaptation** architecture, applied to the **Quadruped** environment using the **domain_randomization** policy.
+
+These checkpoints represent the fully-converged policy at the final epoch (Epoch 600) across 3 independent random seeds (`seed_1`, `seed_2`, `seed_42`).
+
+### Architecture
+The policy consists of a dual-objective architecture:
+1. **Vision Transformer (ViT)**: Acts as the state encoder, augmented with an Auxiliary Physics Distillation Head to predict privileged parameters (like mass or friction).
+2. **1D U-Net Vector Field**: A deterministic, parallelized Optimal Transport Flow Matching (Rectified Flows) model providing $O(1)$ generative inference latency.
+
+### Usage
+The checkpoints are saved in the highly-optimized **JAX / Orbax** format.
+
+```python
+import orbax.checkpoint as ocp
+
+# Example: Loading seed 42
+checkpoint_path = "seed_42"
+mngr = ocp.CheckpointManager(os.path.abspath(checkpoint_path))
+
+# Load the parameters
+restored = mngr.restore(mngr.latest_step())
+print(restored.keys())
+```

seed_1/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1779896574498497343, "commit_timestamp_nsecs": 1779896575245197080, "custom_metadata": {}}

seed_1/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('buffer', 'proprio_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "proprio_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 27]}}, "('buffer', 'privileged_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "privileged_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 6]}}, "('buffer', 'vision_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "vision_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 64, 64, 1]}}, "('buffer', 'action_chunks')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "action_chunks", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 16, 12]}}, "('buffer', 'ptr')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "ptr", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('buffer', 'size')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "size", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('student', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [6]}}, "('student', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 6]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [6]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 6]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 6]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 268, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [12]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 12]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [283, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 8, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'cls_token')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "cls_token", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'pos_embedding')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 65, 128]}}, "('teacher', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [33, 256]}}, "('teacher', 'params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('teacher', 'params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [12]}}, "('teacher', 'params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 12]}}, "('teacher', 'params', 'Dense_3', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1]}}, "('teacher', 'params', 'Dense_3', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1]}}, "('teacher', 'params', 'log_std')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "log_std", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [12]}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

seed_1/default/_sharding

@@ -0,0 +1 @@
+{"YnVmZmVyLmFjdGlvbl9jaHVua3M=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnB0cg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByaXZpbGVnZWRfb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByb3ByaW9fb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnNpemU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnZpc2lvbl9vYnM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLmNsc190b2tlbg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLnBvc19lbWJlZGRpbmc=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMubG9nX3N0ZA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

seed_1/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "buffer.proprio_obs", "write_shape": [100000, 27], "chunk_shape": [100000, 27], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.privileged_obs", "write_shape": [100000, 6], "chunk_shape": [100000, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.vision_obs", "write_shape": [100000, 64, 64, 1], "chunk_shape": [100000, 64, 64, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.action_chunks", "write_shape": [100000, 16, 12], "chunk_shape": [100000, 16, 12], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.ptr", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.size", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.bias", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.kernel", "write_shape": [256, 6], "chunk_shape": [256, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.kernel", "write_shape": [128, 128], "chunk_shape": [128, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_B.kernel", "write_shape": [8, 128], "chunk_shape": [8, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.bias", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.kernel", "write_shape": [128, 6], "chunk_shape": [128, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_B.kernel", "write_shape": [8, 6], "chunk_shape": [8, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.kernel", "write_shape": [3, 268, 256], "chunk_shape": [3, 268, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.kernel", "write_shape": [3, 256, 256], "chunk_shape": [3, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.bias", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.kernel", "write_shape": [3, 256, 12], "chunk_shape": [3, 256, 12], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.kernel", "write_shape": [1, 128], "chunk_shape": [1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.kernel", "write_shape": [283, 128], "chunk_shape": [283, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.kernel", "write_shape": [8, 8, 1, 128], "chunk_shape": [8, 8, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.cls_token", "write_shape": [1, 1, 128], "chunk_shape": [1, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.pos_embedding", "write_shape": [1, 65, 128], "chunk_shape": [1, 65, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.kernel", "write_shape": [33, 256], "chunk_shape": [33, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.kernel", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.bias", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.kernel", "write_shape": [256, 12], "chunk_shape": [256, 12], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.kernel", "write_shape": [256, 1], "chunk_shape": [256, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.log_std", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}]}

seed_1/default/d/15630d628f45ecdce68c3260ffadfd66

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74494e41ae9f22c91dbfe769bce1b75fd72b129a04da81445fb95bc218a69e6b
+size 22750

seed_1/default/ocdbt.process_0/d/1a197b22eec6355786cf6f5d27584820

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16459a9a95b8e04a2c87a69b99525f88f9b2a61069f01f680e74d447a033de1c
+size 1063

seed_1/default/ocdbt.process_0/d/3c21383254212135fa7fc9f68e63fe05

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6348a0cd8d953316e80f64f97c1a7d80c053d193c010ef6c3c4ddabb8f7d177b
+size 1095

seed_1/default/ocdbt.process_0/d/3e6a9d362b460463dff14d2352779dc6

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:106fb72ed44f324c2dedbf1b95746297135b69c203d6a7c6b0433acd4117b4f1
+size 191

seed_1/default/ocdbt.process_0/d/424ba21bae4f073ec1cf058c99edb18b

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0efa976fa72f07bca0620f9ba9bee2dd53f1d9bc81af91035a979c3081cb789d
+size 14209024

seed_1/default/ocdbt.process_0/d/bd22ad91f2f40d72c4aa941bdf3ab49e

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cdd55fc7195a0c39b5271906ba14de8bf2c64e3ea14f8a5b7fe355fb4dfa42f3
+size 69357568

seed_1/default/ocdbt.process_0/d/cd47112bbb46d432bd5cadbdc2808fe8

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a2ebf664083194700594ed1c835820d0b7e20339f237386eaa4945b2f55e23e3
+size 72751

seed_2/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1779888816799053300, "commit_timestamp_nsecs": 1779888817548402912, "custom_metadata": {}}

seed_2/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('buffer', 'proprio_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "proprio_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 27]}}, "('buffer', 'privileged_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "privileged_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 6]}}, "('buffer', 'vision_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "vision_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 64, 64, 1]}}, "('buffer', 'action_chunks')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "action_chunks", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 16, 12]}}, "('buffer', 'ptr')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "ptr", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('buffer', 'size')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "size", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('student', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [6]}}, "('student', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 6]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [6]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 6]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 6]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 268, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [12]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 12]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [283, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 8, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'cls_token')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "cls_token", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'pos_embedding')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 65, 128]}}, "('teacher', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [33, 256]}}, "('teacher', 'params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('teacher', 'params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [12]}}, "('teacher', 'params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 12]}}, "('teacher', 'params', 'Dense_3', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1]}}, "('teacher', 'params', 'Dense_3', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1]}}, "('teacher', 'params', 'log_std')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "log_std", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [12]}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

seed_2/default/_sharding

@@ -0,0 +1 @@
+{"YnVmZmVyLmFjdGlvbl9jaHVua3M=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnB0cg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByaXZpbGVnZWRfb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByb3ByaW9fb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnNpemU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnZpc2lvbl9vYnM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLmNsc190b2tlbg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLnBvc19lbWJlZGRpbmc=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMubG9nX3N0ZA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

seed_2/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "buffer.proprio_obs", "write_shape": [100000, 27], "chunk_shape": [100000, 27], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.privileged_obs", "write_shape": [100000, 6], "chunk_shape": [100000, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.vision_obs", "write_shape": [100000, 64, 64, 1], "chunk_shape": [100000, 64, 64, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.action_chunks", "write_shape": [100000, 16, 12], "chunk_shape": [100000, 16, 12], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.ptr", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.size", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.bias", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.kernel", "write_shape": [256, 6], "chunk_shape": [256, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.kernel", "write_shape": [128, 128], "chunk_shape": [128, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_B.kernel", "write_shape": [8, 128], "chunk_shape": [8, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.bias", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.kernel", "write_shape": [128, 6], "chunk_shape": [128, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_B.kernel", "write_shape": [8, 6], "chunk_shape": [8, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.kernel", "write_shape": [3, 268, 256], "chunk_shape": [3, 268, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.kernel", "write_shape": [3, 256, 256], "chunk_shape": [3, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.bias", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.kernel", "write_shape": [3, 256, 12], "chunk_shape": [3, 256, 12], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.kernel", "write_shape": [1, 128], "chunk_shape": [1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.kernel", "write_shape": [283, 128], "chunk_shape": [283, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.kernel", "write_shape": [8, 8, 1, 128], "chunk_shape": [8, 8, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.cls_token", "write_shape": [1, 1, 128], "chunk_shape": [1, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.pos_embedding", "write_shape": [1, 65, 128], "chunk_shape": [1, 65, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.kernel", "write_shape": [33, 256], "chunk_shape": [33, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.kernel", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.bias", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.kernel", "write_shape": [256, 12], "chunk_shape": [256, 12], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.kernel", "write_shape": [256, 1], "chunk_shape": [256, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.log_std", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}]}

seed_2/default/d/c7cbdd6c7354bb330416300939f8af1f

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83b0a0a60ac4f56bfeb89122fce71a969a332ce4cefc9da55185eb440cdd32bf
+size 22747

seed_2/default/ocdbt.process_0/d/1564394f1496af2a8a9efa71727343e8

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5f431711c26378670e1d12e4b9764f087587403a6df79b04f5c1440fee64018a
+size 2935218

seed_2/default/ocdbt.process_0/d/415d18dbf919f6640538ac44d5f553e6

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:106fb72ed44f324c2dedbf1b95746297135b69c203d6a7c6b0433acd4117b4f1
+size 191

seed_2/default/ocdbt.process_0/d/67c5b24751f784c32cdc208e0ede3845

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16459a9a95b8e04a2c87a69b99525f88f9b2a61069f01f680e74d447a033de1c
+size 1063

seed_2/default/ocdbt.process_0/d/7ad552c166dff68f9f6620024c660e86

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7976cdc40bba04320a9270bf24755f005f1ba98181342c2c2d4fd4bb4596cf6f
+size 81252352

seed_2/default/ocdbt.process_0/d/b0a9a146cbc5d81d879ab1e137abfb20

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e8345f07c3acaf902f8104653d23c88cf4a1d70e89993dd161e03c22b6426198
+size 72748

seed_42/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1779882004734080759, "commit_timestamp_nsecs": 1779882005492208084, "custom_metadata": {}}

seed_42/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('buffer', 'proprio_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "proprio_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 27]}}, "('buffer', 'privileged_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "privileged_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 6]}}, "('buffer', 'vision_obs')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "vision_obs", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 64, 64, 1]}}, "('buffer', 'action_chunks')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "action_chunks", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [100000, 16, 12]}}, "('buffer', 'ptr')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "ptr", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('buffer', 'size')": {"key_metadata": [{"key": "buffer", "key_type": 2}, {"key": "size", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('student', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [6]}}, "('student', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 6]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_0', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_0", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 128]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [6]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'base', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "base", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 6]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_A', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_A", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 8]}}, "('student', 'params', 'PhysicsHead_0', 'LowRankDense_1', 'lora_B', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "PhysicsHead_0", "key_type": 2}, {"key": "LowRankDense_1", "key_type": 2}, {"key": "lora_B", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 6]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 268, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'UNet1D_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 256]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [12]}}, "('student', 'params', 'UNet1D_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 256, 12]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'UNet1D_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "UNet1D_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [283, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8, 8, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_0', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_0", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'LayerNorm_1', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'MlpBlock_0', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'key', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'out', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32, 128]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'query', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'EncoderBlock_1', 'SelfAttention_0', 'value', 'kernel')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "EncoderBlock_1", "key_type": 2}, {"key": "SelfAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 4, 32]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'bias')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'LayerNorm_0', 'scale')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('student', 'params', 'VisionEncoder_0', 'cls_token')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "cls_token", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 1, 128]}}, "('student', 'params', 'VisionEncoder_0', 'pos_embedding')": {"key_metadata": [{"key": "student", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "VisionEncoder_0", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 65, 128]}}, "('teacher', 'params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [33, 256]}}, "('teacher', 'params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('teacher', 'params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('teacher', 'params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [12]}}, "('teacher', 'params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 12]}}, "('teacher', 'params', 'Dense_3', 'bias')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1]}}, "('teacher', 'params', 'Dense_3', 'kernel')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "Dense_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1]}}, "('teacher', 'params', 'log_std')": {"key_metadata": [{"key": "teacher", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "log_std", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [12]}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

seed_42/default/_sharding

@@ -0,0 +1 @@
+{"YnVmZmVyLmFjdGlvbl9jaHVua3M=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnB0cg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByaXZpbGVnZWRfb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnByb3ByaW9fb2Jz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnNpemU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YnVmZmVyLnZpc2lvbl9vYnM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMC5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5iYXNlLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Eua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuUGh5c2ljc0hlYWRfMC5Mb3dSYW5rRGVuc2VfMS5sb3JhX0Iua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8xLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuQ29udl8yLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVU5ldDFEXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18wLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLkxheWVyTm9ybV8xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5rZXkua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5vdXQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkVuY29kZXJCbG9ja18xLlNlbGZBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLmNsc190b2tlbg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3R1ZGVudC5wYXJhbXMuVmlzaW9uRW5jb2Rlcl8wLnBvc19lbWJlZGRpbmc=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMuRGVuc2VfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","dGVhY2hlci5wYXJhbXMubG9nX3N0ZA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

seed_42/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "buffer.proprio_obs", "write_shape": [100000, 27], "chunk_shape": [100000, 27], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.privileged_obs", "write_shape": [100000, 6], "chunk_shape": [100000, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.vision_obs", "write_shape": [100000, 64, 64, 1], "chunk_shape": [100000, 64, 64, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.action_chunks", "write_shape": [100000, 16, 12], "chunk_shape": [100000, 16, 12], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.ptr", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "buffer.size", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.bias", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.Dense_0.kernel", "write_shape": [256, 6], "chunk_shape": [256, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.base.kernel", "write_shape": [128, 128], "chunk_shape": [128, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_0.lora_B.kernel", "write_shape": [8, 128], "chunk_shape": [8, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.bias", "write_shape": [6], "chunk_shape": [6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.base.kernel", "write_shape": [128, 6], "chunk_shape": [128, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_A.kernel", "write_shape": [128, 8], "chunk_shape": [128, 8], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.PhysicsHead_0.LowRankDense_1.lora_B.kernel", "write_shape": [8, 6], "chunk_shape": [8, 6], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_0.kernel", "write_shape": [3, 268, 256], "chunk_shape": [3, 268, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_1.kernel", "write_shape": [3, 256, 256], "chunk_shape": [3, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.bias", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Conv_2.kernel", "write_shape": [3, 256, 12], "chunk_shape": [3, 256, 12], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_0.kernel", "write_shape": [1, 128], "chunk_shape": [1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.UNet1D_0.Dense_1.kernel", "write_shape": [283, 128], "chunk_shape": [283, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Conv_0.kernel", "write_shape": [8, 8, 1, 128], "chunk_shape": [8, 8, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_0.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.LayerNorm_1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.MlpBlock_0.Dense_1.kernel", "write_shape": [256, 128], "chunk_shape": [256, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.key.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.out.kernel", "write_shape": [4, 32, 128], "chunk_shape": [4, 32, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.query.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.bias", "write_shape": [4, 32], "chunk_shape": [4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.EncoderBlock_1.SelfAttention_0.value.kernel", "write_shape": [128, 4, 32], "chunk_shape": [128, 4, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.LayerNorm_0.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.cls_token", "write_shape": [1, 1, 128], "chunk_shape": [1, 1, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "student.params.VisionEncoder_0.pos_embedding", "write_shape": [1, 65, 128], "chunk_shape": [1, 65, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_0.kernel", "write_shape": [33, 256], "chunk_shape": [33, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_1.kernel", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.bias", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_2.kernel", "write_shape": [256, 12], "chunk_shape": [256, 12], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.bias", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.Dense_3.kernel", "write_shape": [256, 1], "chunk_shape": [256, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "teacher.params.log_std", "write_shape": [12], "chunk_shape": [12], "ext_metadata": null}}]}

seed_42/default/d/e3c48eb13d81ebb0aa05bbf7a1946893

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:968166575928077f61e9188c01c1f22eea0af0c2aa3b4b3801ce4ed223a341ba
+size 21765

seed_42/default/ocdbt.process_0/d/1ff52450f39db1503eb1d20860b31911

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:31c1cb29a8f1c86c187cb52ddb9fd113e8bec8cf4a973a412b1ef86ac25857ae
+size 69726208

seed_42/default/ocdbt.process_0/d/7b2fe88b8b7fc8bdf407bb8f705e778a

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16459a9a95b8e04a2c87a69b99525f88f9b2a61069f01f680e74d447a033de1c
+size 1063

seed_42/default/ocdbt.process_0/d/9247c711def78e54ac769a283647ff95

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d0bfa3504ea19c36c6fd2bd5d6333bdd0db8966217b6c6e0d0597e4c5238ab0a
+size 224

seed_42/default/ocdbt.process_0/d/9e3be425cba0f07e6c78bb85928bd9c6

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fe758c088a9874aa5059e05a295c59372364521488d3ad2d7395976e7f33c348
+size 71767

seed_42/default/ocdbt.process_0/d/b85b1c49ecc868e1bdd9a2deaa231400

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4fcc08c2751e2f1dbdecd6030c9890a3f043479aa11edf58fab9855f443ce456
+size 13651968

seed_42/default/ocdbt.process_0/d/c9ecd154d346b2f7aba81d19aa2b7d00

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1a86854c01ed9919f97752d45bf65610f7aff2680237cb9a0e3eba2dcd7f927f
+size 559756