convert regression weights to safetensors
Browse files- regression/config.json +12 -13
- regression/model.safetensors +3 -0
regression/config.json
CHANGED
|
@@ -1,18 +1,17 @@
|
|
| 1 |
{
|
| 2 |
-
"model_type": "tabfm",
|
| 3 |
-
"version": "1.0.0",
|
| 4 |
-
"task": "regression",
|
| 5 |
-
"framework": "pytorch",
|
| 6 |
-
"embed_dim": 256,
|
| 7 |
-
"max_classes": 10,
|
| 8 |
-
"col_num_blocks": 3,
|
| 9 |
"col_nhead": 4,
|
|
|
|
| 10 |
"col_num_inds": 256,
|
| 11 |
-
"
|
| 12 |
-
"
|
| 13 |
-
"
|
| 14 |
-
"icl_num_blocks": 24,
|
| 15 |
-
"icl_nhead": 8,
|
| 16 |
"ff_factor": 4,
|
| 17 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
}
|
|
|
|
| 1 |
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
"col_nhead": 4,
|
| 3 |
+
"col_num_blocks": 3,
|
| 4 |
"col_num_inds": 256,
|
| 5 |
+
"decoder_hidden": null,
|
| 6 |
+
"embed_dim": 256,
|
| 7 |
+
"feature_group_size": 3,
|
|
|
|
|
|
|
| 8 |
"ff_factor": 4,
|
| 9 |
+
"icl_nhead": 8,
|
| 10 |
+
"icl_num_blocks": 24,
|
| 11 |
+
"is_classifier": false,
|
| 12 |
+
"max_classes": 10,
|
| 13 |
+
"num_freq": 32,
|
| 14 |
+
"row_nhead": 8,
|
| 15 |
+
"row_num_blocks": 3,
|
| 16 |
+
"row_num_cls": 8
|
| 17 |
}
|
regression/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bd5a615b0322a8f04a895038de6df6fbd71430eca750e1d792f31048654674a9
|
| 3 |
+
size 6591243724
|