rinkorn commited on
Commit
836096a
1 Parent(s): 7065454

rinkorn/distilbert-base-uncased-finetuned-imdb

Browse files
Files changed (2) hide show
  1. README.md +8 -8
  2. model.safetensors +1 -1
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  license: apache-2.0
3
- base_model: distilbert-base-uncased
4
  tags:
5
  - generated_from_trainer
6
  model-index:
@@ -13,9 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
13
 
14
  # distilbert-base-uncased-finetuned-imdb
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: 5.7313
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: 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,9 +47,9 @@ The following hyperparameters were used during training:
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
- | 6.9693 | 1.0 | 157 | 5.9811 |
51
- | 5.8737 | 2.0 | 314 | 5.7570 |
52
- | 5.7375 | 3.0 | 471 | 5.6725 |
53
 
54
 
55
  ### Framework versions
 
1
  ---
2
  license: apache-2.0
3
+ base_model: Geotrend/distilbert-base-ru-cased
4
  tags:
5
  - generated_from_trainer
6
  model-index:
 
13
 
14
  # distilbert-base-uncased-finetuned-imdb
15
 
16
+ This model is a fine-tuned version of [Geotrend/distilbert-base-ru-cased](https://huggingface.co/Geotrend/distilbert-base-ru-cased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 3.7537
19
 
20
  ## Model description
21
 
 
35
 
36
  The following hyperparameters were used during training:
37
  - learning_rate: 2e-05
38
+ - train_batch_size: 96
39
+ - eval_batch_size: 96
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
+ | 4.1884 | 1.0 | 990 | 3.9081 |
51
+ | 3.9108 | 2.0 | 1980 | 3.7920 |
52
+ | 3.8425 | 3.0 | 2970 | 3.7537 |
53
 
54
 
55
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:544b10c02d53af292327f3e5d029f265542bf0b7089fcc3eb13e87366a4227dc
3
  size 218003556
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b31eae34da4082329c882abd798e1d1b853dd07e7cd2d05e9e1565e1a366ce13
3
  size 218003556