Thenghuy commited on
Commit
569d191
verified
1 Parent(s): ca17636

End of training

Browse files
Files changed (2) hide show
  1. README.md +9 -16
  2. model.safetensors +1 -1
README.md CHANGED
@@ -15,7 +15,12 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [google/electra-base-discriminator](https://huggingface.co/google/electra-base-discriminator) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 1.3494
 
 
 
 
 
19
 
20
  ## Model description
21
 
@@ -35,24 +40,12 @@ More information needed
35
 
36
  The following hyperparameters were used during training:
37
  - learning_rate: 2e-05
38
- - train_batch_size: 16
39
- - eval_batch_size: 16
40
  - seed: 42
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
43
- - num_epochs: 6
44
-
45
- ### Training results
46
-
47
- | Training Loss | Epoch | Step | Validation Loss |
48
- |:-------------:|:-----:|:----:|:---------------:|
49
- | No log | 1.0 | 250 | 1.1834 |
50
- | 1.7393 | 2.0 | 500 | 1.0573 |
51
- | 1.7393 | 3.0 | 750 | 1.0987 |
52
- | 0.5696 | 4.0 | 1000 | 1.1640 |
53
- | 0.5696 | 5.0 | 1250 | 1.3014 |
54
- | 0.3252 | 6.0 | 1500 | 1.3494 |
55
-
56
 
57
  ### Framework versions
58
 
 
15
 
16
  This model is a fine-tuned version of [google/electra-base-discriminator](https://huggingface.co/google/electra-base-discriminator) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - eval_loss: 1.6542
19
+ - eval_runtime: 24.211
20
+ - eval_samples_per_second: 41.304
21
+ - eval_steps_per_second: 6.898
22
+ - epoch: 2.0
23
+ - step: 1334
24
 
25
  ## Model description
26
 
 
40
 
41
  The following hyperparameters were used during training:
42
  - learning_rate: 2e-05
43
+ - train_batch_size: 6
44
+ - eval_batch_size: 6
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
+ - num_epochs: 3
 
 
 
 
 
 
 
 
 
 
 
 
49
 
50
  ### Framework versions
51
 
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:df58f1d001c6676ade70a895015883fc7101c29d0ce7a12d9e4dc86a738b782e
3
  size 435596680
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e66a2f9580ea9e5d28b25e103d85cb1225d37b089ce6b4149f6e947c85cc63cb
3
  size 435596680