Upload folder using huggingface_hub

.gitattributes

@@ -38,3 +38,6 @@ prev_y9-eir/checkpoint/RQS/0/model/ocdbt.process_0/d/8995524c9fdeab3225b3d7759e4
 checkpoint/RQS/0/model/ocdbt.process_0/d/3c40a069fa979a137250b0d0875d8358 filter=lfs diff=lfs merge=lfs -text
 hbr_y9-eir/checkpoint/RQS/0/model/ocdbt.process_0/d/3c40a069fa979a137250b0d0875d8358 filter=lfs diff=lfs merge=lfs -text
 prev_y9-eir/checkpoint/RQS/0/model/ocdbt.process_0/d/bbd25b900aa160674a4e6d0083ca97cd filter=lfs diff=lfs merge=lfs -text
+eir-hbr_y9/checkpoint/RQS/0/model/ocdbt.process_0/d/3dd764421db9ed296fbcb0477abe4b31 filter=lfs diff=lfs merge=lfs -text
+hbr_y9-eir/checkpoint/RQS/0/model/ocdbt.process_0/d/912ebe1633cc0f3ddeb271ff08ac0178 filter=lfs diff=lfs merge=lfs -text
+prev_y9-eir/checkpoint/RQS/0/model/ocdbt.process_0/d/d35ee908ed97bf7f137859ca425be6ad filter=lfs diff=lfs merge=lfs -text

eir-hbr_y9/checkpoint/RQS/0/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"model": "orbax.checkpoint.experimental.v1._src.handlers.pytree_handler.PyTreeHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1786089415169362895, "commit_timestamp_nsecs": 1786089415363375422, "custom_metadata": {}}

eir-hbr_y9/checkpoint/RQS/0/model/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('net', 'dropout', 'rngs', 'count', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "dropout", "key_type": 2}, {"key": "rngs", "key_type": 2}, {"key": "count", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('net', 'dropout', 'rngs', 'key', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "dropout", "key_type": 2}, {"key": "rngs", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('net', 'fc1', '0', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc1', '0', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc1', '1', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc1', '1', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc2', '0', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc2', '0', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc2', '1', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc2', '1', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'inp', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "inp", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'inp', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "inp", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7, 256]}}, "('net', 'norms', '0', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '0', 'scale', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '1', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '1', 'scale', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'out', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [97]}}, "('net', 'out', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 97]}}}, "use_ocdbt": true, "use_zarr3": true, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

eir-hbr_y9/checkpoint/RQS/0/model/_sharding

@@ -0,0 +1 @@
+{"bmV0Lm5vcm1zLjAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm5vcm1zLjAuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm5vcm1zLjEuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm5vcm1zLjEuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm91dC5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm91dC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmRyb3BvdXQucm5ncy5jb3VudC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmRyb3BvdXQucm5ncy5rZXkudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4wLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4xLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4wLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4xLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmlucC5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmlucC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}"}

eir-hbr_y9/checkpoint/RQS/0/model/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "net.dropout.rngs.count.value", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.dropout.rngs.key.value", "write_shape": [2], "chunk_shape": [2], "ext_metadata": {"random_key_impl": "threefry2x32"}}}, {"array_metadata": {"param_name": "net.fc1.0.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.0.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.1.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.1.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.0.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.0.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.1.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.1.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.inp.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.inp.kernel.value", "write_shape": [7, 256], "chunk_shape": [7, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.0.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.0.scale.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.1.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.1.scale.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.out.bias.value", "write_shape": [97], "chunk_shape": [97], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.out.kernel.value", "write_shape": [256, 97], "chunk_shape": [256, 97], "ext_metadata": null}}]}

eir-hbr_y9/checkpoint/RQS/0/model/ocdbt.process_0/d/3dd764421db9ed296fbcb0477abe4b31

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

eir-hbr_y9/config.json

@@ -0,0 +1,47 @@
+{
+  "model_name": "RQS",
+  "predictor": "eir",
+  "target": "hbr_y9",
+  "width": 256,
+  "depth": 2,
+  "n_bins": 32,
+  "rqs_bounds": 6,
+  "mlp_residual": true,
+  "dropout_rate": 0.1,
+  "features": [
+    "eir",
+    "dn0_use",
+    "Q0",
+    "phi_bednets",
+    "seasonal",
+    "itn_use",
+    "irs_use"
+  ],
+  "feature_scalar_mean": [
+    1.1071313619613647,
+    0.3145143687725067,
+    0.7966740131378174,
+    0.6589758992195129,
+    0.469308465719223,
+    0.24106915295124054,
+    0.14334255456924438
+  ],
+  "feature_scalar_scale": [
+    0.796909511089325,
+    0.16059787571430206,
+    0.1178336963057518,
+    0.15951770544052124,
+    0.4990571439266205,
+    0.3166757822036743,
+    0.24969400465488434
+  ],
+  "feature_log_idx": [
+    0
+  ],
+  "target_scalar_mean": [
+    5.388309478759766
+  ],
+  "target_scalar_scale": [
+    0.6828842163085938
+  ]
+}

hbr_y9-eir/checkpoint/RQS/0/_CHECKPOINT_METADATA

@@ -1 +1 @@
-{"item_handlers": {"model": "orbax.checkpoint.experimental.v1._src.handlers.pytree_handler.PyTreeHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1785936908052942591, "commit_timestamp_nsecs": 1785936908253794591, "custom_metadata": {}}
+{"item_handlers": {"model": "orbax.checkpoint.experimental.v1._src.handlers.pytree_handler.PyTreeHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1786089499229937243, "commit_timestamp_nsecs": 1786089499464892961, "custom_metadata": {}}

hbr_y9-eir/checkpoint/RQS/0/model/_METADATA

@@ -1 +1 @@
-{"tree_metadata": {"('net', 'dropout', 'rngs', 'count', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "dropout", "key_type": 2}, {"key": "rngs", "key_type": 2}, {"key": "count", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('net', 'dropout', 'rngs', 'key', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "dropout", "key_type": 2}, {"key": "rngs", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('net', 'fc1', '0', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc1', '0', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc1', '1', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc1', '1', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc1', '2', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc1', '2', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc1', '3', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc1', '3', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc2', '0', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc2', '0', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc2', '1', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc2', '1', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc2', '2', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc2', '2', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc2', '3', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc2', '3', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'inp', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "inp", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'inp', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "inp", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7, 256]}}, "('net', 'norms', '0', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '0', 'scale', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '1', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '1', 'scale', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '2', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '2', 'scale', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '3', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '3', 'scale', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'out', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [97]}}, "('net', 'out', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 97]}}}, "use_ocdbt": true, "use_zarr3": true, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
+{"tree_metadata": {"('net', 'dropout', 'rngs', 'count', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "dropout", "key_type": 2}, {"key": "rngs", "key_type": 2}, {"key": "count", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('net', 'dropout', 'rngs', 'key', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "dropout", "key_type": 2}, {"key": "rngs", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('net', 'fc1', '0', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc1', '0', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc1', '1', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc1', '1', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc1", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc2', '0', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc2', '0', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'fc2', '1', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'fc2', '1', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "fc2", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 256]}}, "('net', 'inp', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "inp", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'inp', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "inp", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7, 256]}}, "('net', 'norms', '0', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '0', 'scale', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '1', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'norms', '1', 'scale', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "norms", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('net', 'out', 'bias', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [97]}}, "('net', 'out', 'kernel', 'value')": {"key_metadata": [{"key": "net", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 97]}}}, "use_ocdbt": true, "use_zarr3": true, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

hbr_y9-eir/checkpoint/RQS/0/model/_sharding

@@ -1 +1 @@
-{"bmV0Lm5vcm1zLjAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm5vcm1zLjAuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm5vcm1zLjEuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm5vcm1zLjEuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm5vcm1zLjIuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm5vcm1zLjIuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm5vcm1zLjMuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm5vcm1zLjMuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm91dC5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm91dC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmRyb3BvdXQucm5ncy5jb3VudC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmRyb3BvdXQucm5ncy5rZXkudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4wLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4xLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4yLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4yLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4zLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4zLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4wLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4xLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4yLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4yLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4zLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4zLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmlucC5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmlucC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
+{"bmV0Lm5vcm1zLjAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm5vcm1zLjAuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm5vcm1zLjEuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm5vcm1zLjEuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm91dC5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm91dC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmRyb3BvdXQucm5ncy5jb3VudC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmRyb3BvdXQucm5ncy5rZXkudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4wLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4xLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4wLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4xLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmlucC5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmlucC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}"}

hbr_y9-eir/checkpoint/RQS/0/model/array_metadatas/process_0

@@ -1 +1 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "net.dropout.rngs.count.value", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.dropout.rngs.key.value", "write_shape": [2], "chunk_shape": [2], "ext_metadata": {"random_key_impl": "threefry2x32"}}}, {"array_metadata": {"param_name": "net.fc1.0.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.0.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.1.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.1.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.2.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.2.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.3.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.3.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.0.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.0.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.1.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.1.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.2.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.2.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.3.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.3.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.inp.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.inp.kernel.value", "write_shape": [7, 256], "chunk_shape": [7, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.0.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.0.scale.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.1.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.1.scale.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.2.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.2.scale.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.3.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.3.scale.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.out.bias.value", "write_shape": [97], "chunk_shape": [97], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.out.kernel.value", "write_shape": [256, 97], "chunk_shape": [256, 97], "ext_metadata": null}}]}
+{"array_metadatas": [{"array_metadata": {"param_name": "net.dropout.rngs.count.value", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.dropout.rngs.key.value", "write_shape": [2], "chunk_shape": [2], "ext_metadata": {"random_key_impl": "threefry2x32"}}}, {"array_metadata": {"param_name": "net.fc1.0.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.0.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.1.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc1.1.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.0.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.0.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.1.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.fc2.1.kernel.value", "write_shape": [256, 256], "chunk_shape": [256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.inp.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.inp.kernel.value", "write_shape": [7, 256], "chunk_shape": [7, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.0.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.0.scale.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.1.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.norms.1.scale.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.out.bias.value", "write_shape": [97], "chunk_shape": [97], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.out.kernel.value", "write_shape": [256, 97], "chunk_shape": [256, 97], "ext_metadata": null}}]}

hbr_y9-eir/checkpoint/RQS/0/model/ocdbt.process_0/d/912ebe1633cc0f3ddeb271ff08ac0178

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

hbr_y9-eir/config.json

@@ -3,9 +3,9 @@
   "predictor": "hbr_y9",
   "target": "eir",
   "width": 256,
-  "depth": 4,
+  "depth": 2,
   "n_bins": 32,
-  "rqs_bounds": 8,
+  "rqs_bounds": 6,
   "mlp_residual": true,
   "dropout_rate": 0.0,
   "features": [
@@ -18,7 +18,7 @@
     "irs_use"
   ],
   "feature_scalar_mean": [
-    870716.0,
+    5.388309478759766,
     0.3145143687725067,
     0.7966740131378174,
     0.6589758992195129,
@@ -27,7 +27,7 @@
     0.14334255456924438
   ],
   "feature_scalar_scale": [
-    1655729.125,
+    0.6828842163085938,
     0.16059787571430206,
     0.1178336963057518,
     0.15951770544052124,
@@ -35,6 +35,9 @@
     0.3166757822036743,
     0.24969400465488434
   ],
+  "feature_log_idx": [
+    0
+  ],
   "target_scalar_mean": [
     1.1071313619613647
   ],

prev_y9-eir/checkpoint/RQS/0/_CHECKPOINT_METADATA

@@ -1 +1 @@
-{"item_handlers": {"model": "orbax.checkpoint.experimental.v1._src.handlers.pytree_handler.PyTreeHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1785936881001777482, "commit_timestamp_nsecs": 1785936881173948545, "custom_metadata": {}}
+{"item_handlers": {"model": "orbax.checkpoint.experimental.v1._src.handlers.pytree_handler.PyTreeHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1786087610954340150, "commit_timestamp_nsecs": 1786087611179677115, "custom_metadata": {}}

prev_y9-eir/checkpoint/RQS/0/model/_sharding

@@ -1 +1 @@
-{"bmV0Lm5vcm1zLjAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm5vcm1zLjAuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm5vcm1zLjEuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm5vcm1zLjEuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm91dC5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0Lm91dC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmRyb3BvdXQucm5ncy5jb3VudC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmRyb3BvdXQucm5ncy5rZXkudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4wLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMS4xLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4wLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmZjMi4xLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmlucC5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bmV0LmlucC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
+{"bmV0Lm5vcm1zLjAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm5vcm1zLjAuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm5vcm1zLjEuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm5vcm1zLjEuc2NhbGUudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm91dC5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0Lm91dC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmRyb3BvdXQucm5ncy5jb3VudC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmRyb3BvdXQucm5ncy5rZXkudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4wLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMS4xLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4wLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmZjMi4xLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmlucC5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}","bmV0LmlucC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cpu:0\"}"}

prev_y9-eir/checkpoint/RQS/0/model/ocdbt.process_0/d/d35ee908ed97bf7f137859ca425be6ad

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

prev_y9-eir/config.json

@@ -5,7 +5,7 @@
   "width": 256,
   "depth": 2,
   "n_bins": 24,
-  "rqs_bounds": 6,
+  "rqs_bounds": 8,
   "mlp_residual": true,
   "dropout_rate": 0.05,
   "features": [
@@ -35,6 +35,7 @@
     0.3166757822036743,
     0.24969400465488434
   ],
+  "feature_log_idx": [],
   "target_scalar_mean": [
     1.1071313619613647
   ],