robingeibel commited on
Commit
7b5075e
1 Parent(s): a209232

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -10
README.md CHANGED
@@ -13,9 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
13
 
14
  # reformer-finetuned-big_patent-16384
15
 
16
- This model is a fine-tuned version of [google/reformer-crime-and-punishment](https://huggingface.co/google/reformer-crime-and-punishment) on the big_patent dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 4.2238
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
@@ -45,16 +45,16 @@ The following hyperparameters were used during training:
45
 
46
  ### Training results
47
 
48
- | Training Loss | Epoch | Step | Validation Loss |
49
- |:-------------:|:-----:|:-----:|:---------------:|
50
- | 4.2727 | 1.0 | 17979 | 4.2608 |
51
- | 4.2425 | 2.0 | 35958 | 4.2382 |
52
- | 4.2341 | 3.0 | 53937 | 4.2238 |
53
 
54
 
55
  ### Framework versions
56
 
57
  - Transformers 4.20.1
58
- - Pytorch 1.12.0+cu102
59
  - Datasets 2.3.2
60
  - Tokenizers 0.12.1
 
13
 
14
  # reformer-finetuned-big_patent-16384
15
 
16
+ This model is a fine-tuned version of [robingeibel/reformer-finetuned-big_patent-16384](https://huggingface.co/robingeibel/reformer-finetuned-big_patent-16384) on the big_patent dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 5.9716
19
 
20
  ## Model description
21
 
 
35
 
36
  The following hyperparameters were used during training:
37
  - learning_rate: 2e-05
38
+ - train_batch_size: 8
39
+ - eval_batch_size: 8
40
  - seed: 42
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
 
45
 
46
  ### Training results
47
 
48
+ | Training Loss | Epoch | Step | Validation Loss |
49
+ |:-------------:|:-----:|:------:|:---------------:|
50
+ | 6.128 | 1.0 | 53286 | 6.1051 |
51
+ | 5.9909 | 2.0 | 106572 | 5.9874 |
52
+ | 5.9542 | 3.0 | 159858 | 5.9716 |
53
 
54
 
55
  ### Framework versions
56
 
57
  - Transformers 4.20.1
58
+ - Pytorch 1.12.0+cu113
59
  - Datasets 2.3.2
60
  - Tokenizers 0.12.1