future_correction_module/params/_METADATA

{"tree_metadata": {"('params', 'action_encoder', 'block', 'attn', 'w_k', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_k", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'attn', 'w_k', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'attn', 'w_o', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_o", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'attn', 'w_o', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_o", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'attn', 'w_q', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_q", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'attn', 'w_q', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'attn', 'w_v', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_v", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'attn', 'w_v', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'ff1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "ff1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'ff1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "ff1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'ff2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "ff2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'ff2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "ff2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'norm1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'norm1', 'scale')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'norm2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'block', 'norm2', 'scale')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "block", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'in_norm', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "in_norm", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'in_norm', 'scale')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "in_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'in_proj', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "in_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'in_proj', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'out_proj', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "out_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'action_encoder', 'out_proj', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_encoder", "key_type": 2}, {"key": "out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'attn', 'w_k', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_k", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'attn', 'w_k', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'attn', 'w_o', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_o", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'attn', 'w_o', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_o", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'attn', 'w_q', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_q", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'attn', 'w_q', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'attn', 'w_v', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_v", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'attn', 'w_v', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'ff1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "ff1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'ff1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "ff1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'ff2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "ff2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'ff2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "ff2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'norm1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'norm1', 'scale')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'norm2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block0', 'norm2', 'scale')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'attn', 'w_k', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_k", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'attn', 'w_k', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'attn', 'w_o', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_o", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'attn', 'w_o', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_o", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'attn', 'w_q', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_q", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'attn', 'w_q', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'attn', 'w_v', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_v", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'attn', 'w_v', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'ff1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "ff1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'ff1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "ff1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'ff2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "ff2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'ff2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "ff2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'norm1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'norm1', 'scale')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'norm2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'block1', 'norm2', 'scale')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "block1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'film', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "film", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'film', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "film", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'logvar_head', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "logvar_head", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'logvar_head', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "logvar_head", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'mean_head', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "mean_head", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'mean_head', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "mean_head", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'u_proj', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "u_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'future_head', 'u_proj', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "future_head", "key_type": 2}, {"key": "u_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'proprio_encoder', 'fc1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "proprio_encoder", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'proprio_encoder', 'fc1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "proprio_encoder", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'proprio_encoder', 'fc2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "proprio_encoder", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'proprio_encoder', 'fc2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "proprio_encoder", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'time_encoder', 'fc1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'time_encoder', 'fc1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'time_encoder', 'fc2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'time_encoder', 'fc2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'attn', 'w_k', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_k", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'attn', 'w_k', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'attn', 'w_o', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_o", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'attn', 'w_o', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_o", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'attn', 'w_q', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_q", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'attn', 'w_q', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'attn', 'w_v', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_v", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'attn', 'w_v', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "w_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'ff1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "ff1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'ff1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "ff1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'ff2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "ff2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'ff2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "ff2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'kv_proj', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "kv_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'kv_proj', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "kv_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'norm', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'norm', 'scale')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('params', 'token_reducer', 'queries')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "token_reducer", "key_type": 2}, {"key": "queries", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}