Upload folder using huggingface_hub

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* 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

prev_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": 1785925647762129162, "commit_timestamp_nsecs": 1785925647944985191, "custom_metadata": {}}

prev_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', '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": [73]}}, "('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, 73]}}}, "use_ocdbt": true, "use_zarr3": true, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

prev_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\"}","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\"}"}

prev_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.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": [73], "chunk_shape": [73], "ext_metadata": null}}, {"array_metadata": {"param_name": "net.out.kernel.value", "write_shape": [256, 73], "chunk_shape": [256, 73], "ext_metadata": null}}]}

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

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

prev_y9-eir/config.json

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