Upload STGFORMER_BS200_XAVIER model trained on METR-LA
Browse files- config.json +7 -0
- metadata.json +1 -1
- model.safetensors +1 -1
config.json
CHANGED
|
@@ -21,5 +21,12 @@
|
|
| 21 |
"sparsity_k": null,
|
| 22 |
"propagation_mode": "power",
|
| 23 |
"temporal_mode": "transformer",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
"use_zero_init": false
|
| 25 |
}
|
|
|
|
| 21 |
"sparsity_k": null,
|
| 22 |
"propagation_mode": "power",
|
| 23 |
"temporal_mode": "transformer",
|
| 24 |
+
"mamba_d_state": 16,
|
| 25 |
+
"mamba_d_conv": 4,
|
| 26 |
+
"mamba_expand": 2,
|
| 27 |
+
"tcn_num_layers": 3,
|
| 28 |
+
"tcn_kernel_size": 3,
|
| 29 |
+
"tcn_dilation_base": 2,
|
| 30 |
+
"tcn_dropout": 0.1,
|
| 31 |
"use_zero_init": false
|
| 32 |
}
|
metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"dataset": "METR-LA",
|
| 3 |
-
"upload_date": "2025-12-
|
| 4 |
"metrics": {},
|
| 5 |
"framework": "PyTorch",
|
| 6 |
"model_type": "STGFORMER_BS200_XAVIER"
|
|
|
|
| 1 |
{
|
| 2 |
"dataset": "METR-LA",
|
| 3 |
+
"upload_date": "2025-12-09T23:35:56.680308",
|
| 4 |
"metrics": {},
|
| 5 |
"framework": "PyTorch",
|
| 6 |
"model_type": "STGFORMER_BS200_XAVIER"
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7231112
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1dbc2c1c315e8fc9823a2e64fc1043c0e45b790487fe7bb5f295170737de42dd
|
| 3 |
size 7231112
|