Nbeau commited on
Commit
d272f9b
·
verified ·
1 Parent(s): 884b5e7

End of training

Browse files
Files changed (3) hide show
  1. README.md +8 -7
  2. generation_config.json +1 -1
  3. model.safetensors +1 -1
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 [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 0.0735
19
 
20
  ## Model description
21
 
@@ -42,18 +42,19 @@ The following hyperparameters were used during training:
42
  - total_train_batch_size: 64
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: constant_with_warmup
45
- - num_epochs: 5
46
  - mixed_precision_training: Native AMP
47
 
48
  ### Training results
49
 
50
  | Training Loss | Epoch | Step | Validation Loss |
51
  |:-------------:|:-----:|:----:|:---------------:|
52
- | 0.2855 | 1.0 | 125 | 0.1888 |
53
- | 0.1262 | 2.0 | 250 | 0.1072 |
54
- | 0.0917 | 3.0 | 375 | 0.0873 |
55
- | 0.0815 | 4.0 | 500 | 0.0761 |
56
- | 0.0785 | 5.0 | 625 | 0.0735 |
 
57
 
58
 
59
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.1254
19
 
20
  ## Model description
21
 
 
42
  - total_train_batch_size: 64
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: constant_with_warmup
45
+ - num_epochs: 6
46
  - mixed_precision_training: Native AMP
47
 
48
  ### Training results
49
 
50
  | Training Loss | Epoch | Step | Validation Loss |
51
  |:-------------:|:-----:|:----:|:---------------:|
52
+ | 0.144 | 1.0 | 125 | 0.1199 |
53
+ | 0.0896 | 2.0 | 250 | 0.0838 |
54
+ | 0.0861 | 3.0 | 375 | 0.0792 |
55
+ | 0.0781 | 4.0 | 500 | 0.0739 |
56
+ | 0.0772 | 5.0 | 625 | 0.0753 |
57
+ | 0.1549 | 6.0 | 750 | 0.1254 |
58
 
59
 
60
  ### Framework versions
generation_config.json CHANGED
@@ -2,6 +2,6 @@
2
  "_from_model_config": true,
3
  "bos_token_id": 1,
4
  "eos_token_id": 2,
5
- "pad_token_id": 2,
6
  "transformers_version": "4.40.2"
7
  }
 
2
  "_from_model_config": true,
3
  "bos_token_id": 1,
4
  "eos_token_id": 2,
5
+ "pad_token_id": 32000,
6
  "transformers_version": "4.40.2"
7
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:559704e9da8d819c84387e547a1e42ee323924e58074bb0600c77ae442cec227
3
  size 441691776
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:993012c6bc1b09893ac6ca245a6af4483a906120bdfa53fdab18123ab4c529cc
3
  size 441691776