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_reversible_flow_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 **reversible_flow** 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": 1779889338531571620, "commit_timestamp_nsecs": 1779889339255769268, "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/b63c8731fdc1152a477c49e282e9c52c

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

seed_1/default/ocdbt.process_0/d/078e70b3092a614d9ff206660315342a

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

seed_1/default/ocdbt.process_0/d/2f731c07adcc7955f06e15a4e69dc1d5

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

seed_1/default/ocdbt.process_0/d/41bff651719dc1efcd40e2e0223a006c

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

seed_1/default/ocdbt.process_0/d/81d5c3bcc52d2178acd7b986d1b8acba

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

seed_1/default/ocdbt.process_0/d/b016f143f94107c81b937875f644a8b1

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

seed_1/default/ocdbt.process_0/d/e808f99685b95fb06a7a717d59d4c802

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

seed_2/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1779881513578747226, "commit_timestamp_nsecs": 1779881514290341439, "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/cd5b7048d37776122b27aee9c25e1d6b

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

seed_2/default/ocdbt.process_0/d/143d77fe4f4ceb8abf35c3532329b97b

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

seed_2/default/ocdbt.process_0/d/1d8677d690cf0c3da32065ba3c068043

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

seed_2/default/ocdbt.process_0/d/562852e4745d72245310c8682fd5cad0

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

seed_2/default/ocdbt.process_0/d/e1484b087e8e49c954aefb9e161a5e6e

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

seed_2/default/ocdbt.process_0/d/e341d05db76b7f940c6db7dbda4f4040

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

seed_42/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1779874674558263385, "commit_timestamp_nsecs": 1779874675440478670, "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/84313bc3ed0caa48c58820704124c836

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

seed_42/default/ocdbt.process_0/d/091ee96b788bccc3b56a11a99d9fb20b

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

seed_42/default/ocdbt.process_0/d/2a738608702ec252ebe7c180b4904832

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

seed_42/default/ocdbt.process_0/d/69adc315d7ba02fb88124274ce9c4589

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

seed_42/default/ocdbt.process_0/d/9963764bb27adfa7550659759bc3117b

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

seed_42/default/ocdbt.process_0/d/baaa99f823a9a798bdb88b3978f486ea

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