tripathysagar commited on
Commit
05bd2b7
1 Parent(s): 2a82575

End of training

Browse files
README.md CHANGED
@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 2.7709
19
 
20
  ## Model description
21
 
@@ -35,8 +35,8 @@ More information needed
35
 
36
  The following hyperparameters were used during training:
37
  - learning_rate: 2e-05
38
- - train_batch_size: 128
39
- - eval_batch_size: 128
40
  - seed: 42
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
@@ -47,11 +47,11 @@ The following hyperparameters were used during training:
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
- | 3.1073 | 1.0 | 79 | 2.8415 |
51
- | 3.0788 | 2.0 | 158 | 2.8331 |
52
- | 3.076 | 3.0 | 237 | 2.8173 |
53
- | 3.081 | 4.0 | 316 | 2.7946 |
54
- | 3.0791 | 5.0 | 395 | 2.8317 |
55
 
56
 
57
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 2.8476
19
 
20
  ## Model description
21
 
 
35
 
36
  The following hyperparameters were used during training:
37
  - learning_rate: 2e-05
38
+ - train_batch_size: 64
39
+ - eval_batch_size: 64
40
  - seed: 42
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
 
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
+ | 3.0412 | 1.0 | 313 | 2.8585 |
51
+ | 3.019 | 2.0 | 626 | 2.8627 |
52
+ | 2.9974 | 3.0 | 939 | 2.8508 |
53
+ | 2.991 | 4.0 | 1252 | 2.8339 |
54
+ | 2.9919 | 5.0 | 1565 | 2.8476 |
55
 
56
 
57
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:690745c405232e80d7fea925f812c0784b62e10f8cd2141b57434d5ddb888aef
3
  size 267954768
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86c1d2df55ad31e0fca0d2d760495bd8c36664efec2b1c5b48a94b5a9c42c3d1
3
  size 267954768
runs/Feb25_17-39-51_8cfa00f585f9/events.out.tfevents.1708882798.8cfa00f585f9.161.10 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:93b01f16de19abc8d7d39e4cb36572a2e8dc8af6d6edde514b67c3865a6f7607
3
- size 6244
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1e9b9057842613fb54bad684e44db162760883abdaa7cc72bd546d602957fed
3
+ size 7026