witgaw commited on
Commit
74c5f91
·
verified ·
1 Parent(s): 61efd44

Upload STGFORMER_BS200_XAVIER model trained on METR-LA

Browse files
Files changed (3) hide show
  1. config.json +7 -0
  2. metadata.json +1 -1
  3. 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-06T03:19:08.320472",
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:7275e0c2ca7b2bd70548f3001f1245211a1ffa71c1b9254a682ca86d876d89cb
3
  size 7231112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1dbc2c1c315e8fc9823a2e64fc1043c0e45b790487fe7bb5f295170737de42dd
3
  size 7231112