osanseviero
commited on
Commit
•
729c4da
1
Parent(s):
bc9d0c3
End of training
Browse files- README.md +7 -4
- pytorch_model.bin +1 -1
README.md
CHANGED
@@ -17,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
17 |
|
18 |
This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the ag_news dataset.
|
19 |
It achieves the following results on the evaluation set:
|
20 |
-
- Loss: 3.
|
21 |
|
22 |
## Model description
|
23 |
|
@@ -48,9 +48,12 @@ The following hyperparameters were used during training:
|
|
48 |
|
49 |
| Training Loss | Epoch | Step | Validation Loss |
|
50 |
|:-------------:|:-----:|:----:|:---------------:|
|
51 |
-
| 3.
|
52 |
-
|
|
53 |
-
|
|
|
|
|
|
|
|
54 |
|
55 |
|
56 |
### Framework versions
|
|
|
17 |
|
18 |
This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the ag_news dataset.
|
19 |
It achieves the following results on the evaluation set:
|
20 |
+
- Loss: 3.3980
|
21 |
|
22 |
## Model description
|
23 |
|
|
|
48 |
|
49 |
| Training Loss | Epoch | Step | Validation Loss |
|
50 |
|:-------------:|:-----:|:----:|:---------------:|
|
51 |
+
| 3.7271 | 0.32 | 200 | 3.6065 |
|
52 |
+
| 3.346 | 0.64 | 400 | 3.4732 |
|
53 |
+
| 3.0685 | 0.96 | 600 | 3.3985 |
|
54 |
+
| 2.1435 | 1.28 | 800 | 3.4433 |
|
55 |
+
| 1.9834 | 1.6 | 1000 | 3.4203 |
|
56 |
+
| 1.8937 | 1.92 | 1200 | 3.3980 |
|
57 |
|
58 |
|
59 |
### Framework versions
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 497807197
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ebad7d4e7fd700be342c7db7e91c93b7e35af759d45a4945f85fab4206a1055a
|
3 |
size 497807197
|