igor
commited on
End of training
Browse files- README.md +15 -10
- config.json +1 -2
- model.safetensors +1 -1
- training_args.bin +1 -1
README.md
CHANGED
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
|
|
18 |
|
19 |
This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
|
20 |
It achieves the following results on the evaluation set:
|
21 |
-
- Loss: 1.
|
22 |
-
- F1: 0.
|
23 |
|
24 |
## Model description
|
25 |
|
@@ -44,17 +44,22 @@ The following hyperparameters were used during training:
|
|
44 |
- seed: 42
|
45 |
- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
46 |
- lr_scheduler_type: linear
|
47 |
-
- num_epochs:
|
48 |
|
49 |
### Training results
|
50 |
|
51 |
-
| Training Loss | Epoch | Step
|
52 |
-
|
53 |
-
| 0.
|
54 |
-
| 0.
|
55 |
-
| 0.
|
56 |
-
| 0.
|
57 |
-
| 0.
|
|
|
|
|
|
|
|
|
|
|
58 |
|
59 |
|
60 |
### Framework versions
|
|
|
18 |
|
19 |
This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
|
20 |
It achieves the following results on the evaluation set:
|
21 |
+
- Loss: 1.2189
|
22 |
+
- F1: 0.6287
|
23 |
|
24 |
## Model description
|
25 |
|
|
|
44 |
- seed: 42
|
45 |
- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
46 |
- lr_scheduler_type: linear
|
47 |
+
- num_epochs: 10
|
48 |
|
49 |
### Training results
|
50 |
|
51 |
+
| Training Loss | Epoch | Step | Validation Loss | F1 |
|
52 |
+
|:-------------:|:-----:|:-----:|:---------------:|:------:|
|
53 |
+
| 0.6602 | 1.0 | 1367 | 0.6489 | 0.6277 |
|
54 |
+
| 0.6232 | 2.0 | 2734 | 0.6546 | 0.6535 |
|
55 |
+
| 0.5985 | 3.0 | 4101 | 0.7058 | 0.6510 |
|
56 |
+
| 0.5515 | 4.0 | 5468 | 0.6955 | 0.6478 |
|
57 |
+
| 0.505 | 5.0 | 6835 | 0.7862 | 0.6461 |
|
58 |
+
| 0.4617 | 6.0 | 8202 | 0.8992 | 0.6470 |
|
59 |
+
| 0.4538 | 7.0 | 9569 | 1.0468 | 0.6330 |
|
60 |
+
| 0.3878 | 8.0 | 10936 | 1.0613 | 0.6311 |
|
61 |
+
| 0.3665 | 9.0 | 12303 | 1.1309 | 0.6261 |
|
62 |
+
| 0.3483 | 10.0 | 13670 | 1.2189 | 0.6287 |
|
63 |
|
64 |
|
65 |
### Framework versions
|
config.json
CHANGED
@@ -6,7 +6,7 @@
|
|
6 |
],
|
7 |
"attention_dropout": 0.1,
|
8 |
"dim": 768,
|
9 |
-
"dropout": 0.
|
10 |
"hidden_dim": 3072,
|
11 |
"initializer_range": 0.02,
|
12 |
"max_position_embeddings": 512,
|
@@ -18,7 +18,6 @@
|
|
18 |
"qa_dropout": 0.1,
|
19 |
"seq_classif_dropout": 0.2,
|
20 |
"sinusoidal_pos_embds": false,
|
21 |
-
"tie_weights_": true,
|
22 |
"torch_dtype": "float32",
|
23 |
"transformers_version": "4.46.3",
|
24 |
"vocab_size": 30522
|
|
|
6 |
],
|
7 |
"attention_dropout": 0.1,
|
8 |
"dim": 768,
|
9 |
+
"dropout": 0.5,
|
10 |
"hidden_dim": 3072,
|
11 |
"initializer_range": 0.02,
|
12 |
"max_position_embeddings": 512,
|
|
|
18 |
"qa_dropout": 0.1,
|
19 |
"seq_classif_dropout": 0.2,
|
20 |
"sinusoidal_pos_embds": false,
|
|
|
21 |
"torch_dtype": "float32",
|
22 |
"transformers_version": "4.46.3",
|
23 |
"vocab_size": 30522
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 267832560
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:def1f7c6187f7f65dc0c43f6f35b2f47ba5429a3ad017d1499b81a68cd1dcff8
|
3 |
size 267832560
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 5240
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4540b317871d86be6385a2e2a6aee44be3b0dac30b22b816003e3e3588b1b47c
|
3 |
size 5240
|