Sarmila
/

scibert_scivocab_uncased_ml-finetuned-imdb

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

Sarmila commited on Mar 27, 2023

Commit

81e6df6

·

1 Parent(s): 2f89142

update model card README.md

Files changed (1) hide show

README.md +5 -5

README.md CHANGED Viewed

@@ -13,7 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [allenai/scibert_scivocab_uncased](https://huggingface.co/allenai/scibert_scivocab_uncased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 7.0814
 ## Model description
@@ -32,7 +32,7 @@ More information needed
 ### Training hyperparameters
 The following hyperparameters were used during training:
-- learning_rate: 0.002
 - train_batch_size: 32
 - eval_batch_size: 32
 - seed: 42
@@ -45,9 +45,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 8.5111        | 1.0   | 60   | 7.4064          |
-| 7.1522        | 2.0   | 120  | 7.1445          |
-| 7.0532        | 3.0   | 180  | 7.1501          |
 ### Framework versions

 This model is a fine-tuned version of [allenai/scibert_scivocab_uncased](https://huggingface.co/allenai/scibert_scivocab_uncased) on the None dataset.
 It achieves the following results on the evaluation set:
+- Loss: 2.4109
 ## Model description
 ### Training hyperparameters
 The following hyperparameters were used during training:
+- learning_rate: 2e-05
 - train_batch_size: 32
 - eval_batch_size: 32
 - seed: 42
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
+| 2.6232        | 1.0   | 60   | 2.3854          |
+| 2.5312        | 2.0   | 120  | 2.3899          |
+| 2.4464        | 3.0   | 180  | 2.3915          |
 ### Framework versions