Upload folder using huggingface_hub

.gitattributes

@@ -36,3 +36,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 eir-prev_y9/checkpoint/RQS/0/model/ocdbt.process_0/d/ce45e50603f6e146ac07c2b07f12a957 filter=lfs diff=lfs merge=lfs -text
 prev_y9-eir/checkpoint/RQS/0/model/ocdbt.process_0/d/8995524c9fdeab3225b3d7759e429c08 filter=lfs diff=lfs merge=lfs -text
 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

hbr_y9-eir/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": 1785936908052942591, "commit_timestamp_nsecs": 1785936908253794591, "custom_metadata": {}}

hbr_y9-eir/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', '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}

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

@@ -0,0 +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\"}"}

hbr_y9-eir/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.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}}]}

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

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

hbr_y9-eir/config.json

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