Upload pretrained checkpoint
Browse files- imputation_head.safetensors +1 -1
- metadata.json +1 -1
- model.safetensors +1 -1
- pretrain_config.json +4 -3
imputation_head.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 6328
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0ce960f5aa11c25bbc7e2be97626eea79005df8b42e68a51b7fc1d17e7cc7390
|
| 3 |
size 6328
|
metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"dataset": "PEMS-BAY",
|
| 3 |
-
"upload_date": "2025-12-
|
| 4 |
"metrics": {},
|
| 5 |
"framework": "PyTorch",
|
| 6 |
"model_type": "STGFORMER_PRETRAINED"
|
|
|
|
| 1 |
{
|
| 2 |
"dataset": "PEMS-BAY",
|
| 3 |
+
"upload_date": "2025-12-10T09:34:20.480125",
|
| 4 |
"metrics": {},
|
| 5 |
"framework": "PyTorch",
|
| 6 |
"model_type": "STGFORMER_PRETRAINED"
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 7684240
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6b34a02c3df1f6aec9c9b90075b7058c3336ebcd9271d57be0093b9f92ce0d6b
|
| 3 |
size 7684240
|
pretrain_config.json
CHANGED
|
@@ -5,10 +5,11 @@
|
|
| 5 |
"stage1_mask_ratio": 0.15,
|
| 6 |
"stage2_epochs": 5,
|
| 7 |
"stage2_mask_ratio": 0.1,
|
| 8 |
-
"
|
| 9 |
"pretrain_batch_size": 100,
|
| 10 |
-
"
|
| 11 |
-
"save_to": "STGFORMER_PRETRAINED"
|
|
|
|
| 12 |
},
|
| 13 |
"model_dim": 128,
|
| 14 |
"num_nodes": 325
|
|
|
|
| 5 |
"stage1_mask_ratio": 0.15,
|
| 6 |
"stage2_epochs": 5,
|
| 7 |
"stage2_mask_ratio": 0.1,
|
| 8 |
+
"use_normalized_data": false,
|
| 9 |
"pretrain_batch_size": 100,
|
| 10 |
+
"learning_rate": 0.001,
|
| 11 |
+
"save_to": "STGFORMER_PRETRAINED",
|
| 12 |
+
"imputation_iterations": 3
|
| 13 |
},
|
| 14 |
"model_dim": 128,
|
| 15 |
"num_nodes": 325
|