Upload model
Browse files- config.json +5 -3
- model.safetensors +2 -2
config.json
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
{
|
| 2 |
-
"_name_or_path": "checkpoint-1400",
|
| 3 |
"architectures": [
|
| 4 |
-
"
|
| 5 |
],
|
| 6 |
"attention_probs_dropout_prob": 0.1,
|
|
|
|
| 7 |
"classifier_dropout": null,
|
| 8 |
"finetuning_task": "text-classification",
|
| 9 |
"hidden_act": "gelu",
|
|
@@ -55,9 +56,10 @@
|
|
| 55 |
],
|
| 56 |
"layer_norm_eps": 1e-12,
|
| 57 |
"max_position_embeddings": 512,
|
| 58 |
-
"model_type": "
|
| 59 |
"num_attention_heads": 12,
|
| 60 |
"num_hidden_layers": 12,
|
|
|
|
| 61 |
"pad_token_id": 0,
|
| 62 |
"position_embedding_type": "absolute",
|
| 63 |
"problem_types": [
|
|
|
|
| 1 |
{
|
| 2 |
+
"_name_or_path": "/scratch/rtouchen/test_multi_task/checkpoint-1400",
|
| 3 |
"architectures": [
|
| 4 |
+
"MultiTaskClsModel"
|
| 5 |
],
|
| 6 |
"attention_probs_dropout_prob": 0.1,
|
| 7 |
+
"attn_implementation": null,
|
| 8 |
"classifier_dropout": null,
|
| 9 |
"finetuning_task": "text-classification",
|
| 10 |
"hidden_act": "gelu",
|
|
|
|
| 56 |
],
|
| 57 |
"layer_norm_eps": 1e-12,
|
| 58 |
"max_position_embeddings": 512,
|
| 59 |
+
"model_type": "multitaskcls",
|
| 60 |
"num_attention_heads": 12,
|
| 61 |
"num_hidden_layers": 12,
|
| 62 |
+
"num_tasks": 3,
|
| 63 |
"pad_token_id": 0,
|
| 64 |
"position_embedding_type": "absolute",
|
| 65 |
"problem_types": [
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:233ac4b1f64583c194981c3649607d6787b5a3282cf21309941ab01504595ba8
|
| 3 |
+
size 437977480
|