Upload pretrained checkpoint
Browse files- imputation_head.safetensors +1 -1
- metadata.json +1 -1
- model.safetensors +1 -1
- pretrain_config.json +0 -1
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:49d1eabc0235e2b0bd299e16eb790c952e7b3dc62cd9d6d8bd860e4f7d3c3d18
|
| 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-08T15:43:06.743855",
|
| 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:87a71d6851247dba9ce01dba3977b9fe3c260042e1ee0528cd51242f1d5d4c88
|
| 3 |
size 7684240
|
pretrain_config.json
CHANGED
|
@@ -5,7 +5,6 @@
|
|
| 5 |
"stage1_mask_ratio": 0.15,
|
| 6 |
"stage2_epochs": 5,
|
| 7 |
"stage2_mask_ratio": 0.1,
|
| 8 |
-
"pretrain_window": 12,
|
| 9 |
"learning_rate": 0.001,
|
| 10 |
"pretrain_batch_size": 100,
|
| 11 |
"imputation_iterations": 3,
|
|
|
|
| 5 |
"stage1_mask_ratio": 0.15,
|
| 6 |
"stage2_epochs": 5,
|
| 7 |
"stage2_mask_ratio": 0.1,
|
|
|
|
| 8 |
"learning_rate": 0.001,
|
| 9 |
"pretrain_batch_size": 100,
|
| 10 |
"imputation_iterations": 3,
|