Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+eir-prev_y9/checkpoint/RQS/0/model/ocdbt.process_0/d/ce45e50603f6e146ac07c2b07f12a957 filter=lfs diff=lfs merge=lfs -text

eir-prev_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": 1785919135120669525, "commit_timestamp_nsecs": 1785919135238179452, "custom_metadata": {}}

eir-prev_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": [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}

eir-prev_y9/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\"}"}

eir-prev_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": [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}}]}

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

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

eir-prev_y9/config.json

@@ -0,0 +1,35 @@
+{
+  "model_name": "RQS",
+  "predictor": "eir",
+  "target": "prev_y9",
+  "width": 256,
+  "depth": 2,
+  "n_bins": 24,
+  "rqs_bounds": 6,
+  "mlp_residual": true,
+  "dropout_rate": 0.05,
+  "feature_scalar_mean": [
+    57.44230651855469,
+    0.3145143687725067,
+    0.7966740131378174,
+    0.6589758992195129,
+    0.469308465719223,
+    0.24106915295124054,
+    0.14334255456924438
+  ],
+  "feature_scalar_scale": [
+    103.89226531982422,
+    0.16059787571430206,
+    0.1178336963057518,
+    0.15951770544052124,
+    0.4990571439266205,
+    0.3166757822036743,
+    0.24969400465488434
+  ],
+  "target_scalar_mean": [
+    -0.731735110282898
+  ],
+  "target_scalar_scale": [
+    0.46388423442840576
+  ]
+}