End of training

README.md

@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [ashield-ai/prompt-classification-bert](https://huggingface.co/ashield-ai/prompt-classification-bert) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0135
-- F1: 0.9985
+- Loss: nan
+- F1: 0.4955
 
 ## Model description
 
@@ -42,7 +42,7 @@ The following hyperparameters were used during training:
 - train_batch_size: 16
 - eval_batch_size: 8
 - seed: 42
-- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
+- optimizer: Use adamw_torch_fused with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 100
 - num_epochs: 5
@@ -52,16 +52,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step  | Validation Loss | F1     |
 |:-------------:|:-----:|:-----:|:---------------:|:------:|
-| 0.0163        | 1.0   | 2613  | 0.0074          | 0.9989 |
-| 0.0113        | 2.0   | 5226  | 0.0108          | 0.9987 |
-| 0.0047        | 3.0   | 7839  | 0.0147          | 0.9978 |
-| 0.0039        | 4.0   | 10452 | 0.0127          | 0.9983 |
-| 0.0016        | 5.0   | 13065 | 0.0135          | 0.9985 |
+| 0.0           | 1.0   | 6532  | nan             | 0.4955 |
+| 0.0           | 2.0   | 13064 | nan             | 0.4955 |
+| 0.0           | 3.0   | 19596 | nan             | 0.4955 |
+| 0.0           | 4.0   | 26128 | nan             | 0.4955 |
+| 0.0           | 5.0   | 32660 | nan             | 0.4955 |
 
 
 ### Framework versions
 
 - Transformers 4.48.0
-- Pytorch 2.5.1+cu121
+- Pytorch 2.4.0+cu121
 - Datasets 3.2.0
 - Tokenizers 0.21.0

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7141a831c66b3cd11bf8f6bb1b3d7ae320a3a8ea549142c14ddacade5d69db88
+oid sha256:141228a06fa98acb0fb4174bdf87281b38b3a2e6437444da9a170d2985a08327
 size 598439784

runs/Jan17_20-07-12_ip-172-16-109-233.ec2.internal/events.out.tfevents.1737144439.ip-172-16-109-233.ec2.internal.18870.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d7d6524eb28990f36d8e1612fa10f47811246a0fa901d3586aeaf859a2916f05
+size 6195

runs/Jan17_20-25-50_ip-172-16-109-233.ec2.internal/events.out.tfevents.1737145563.ip-172-16-109-233.ec2.internal.18870.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e0f49b7e6535b9402e9aa90d3ecc23c03c656e19f76860bde34973670f835557
+size 5984

runs/Jan17_20-40-11_ip-172-16-109-233.ec2.internal/events.out.tfevents.1737146418.ip-172-16-109-233.ec2.internal.18719.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1adee0381e4244cc1b1e59c0606bfedd856e6fe28c7e7dafee58bfcf124711ee
+size 6406

runs/Jan17_22-25-02_ip-172-16-172-220.ec2.internal/events.out.tfevents.1737152718.ip-172-16-172-220.ec2.internal.8820.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4a77325d0adf4325a3563d2c70467b1b5d95618092324c09687d85973cb2b5cc
+size 24024

runs/Jan18_03-58-29_ip-172-16-172-220.ec2.internal/events.out.tfevents.1737172710.ip-172-16-172-220.ec2.internal.11469.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:499befb25742300e637c49e09e5b965f7d797a4c726ba15f5510af79aabc6de9
+size 5984

runs/Jan18_04-01-30_ip-172-16-172-220.ec2.internal/events.out.tfevents.1737172907.ip-172-16-172-220.ec2.internal.15310.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:316c321b4faeaa2cae37896ea8f19be7f7d371ebf76fe7996c334078845b174e
+size 21794

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cf6a8ff6a52f7c3af45bb8a10acdd828c0220b14a547d2c66b79d859e6d1583c
+oid sha256:d00ce039326b49c04de449071c15141e5f778f69f3e983f895327973407c41b5
 size 5432