betteib commited on
Commit
124534b
1 Parent(s): 7a109d4

Model save

Browse files
Files changed (2) hide show
  1. README.md +9 -11
  2. generation_config.json +1 -1
README.md CHANGED
@@ -1,5 +1,4 @@
1
  ---
2
- library_name: transformers
3
  license: mit
4
  base_model: gpt2
5
  tags:
@@ -12,11 +11,12 @@ model-index:
12
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
  should probably proofread and complete it, then remove this comment. -->
14
 
 
15
  # tn_bacem_gpt2
16
 
17
  This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 4.1384
20
 
21
  ## Model description
22
 
@@ -44,21 +44,19 @@ The following hyperparameters were used during training:
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: cosine
46
  - lr_scheduler_warmup_steps: 500
47
- - num_epochs: 5
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss |
52
  |:-------------:|:------:|:----:|:---------------:|
53
- | 7.3317 | 1.0672 | 500 | 5.1399 |
54
- | 4.6653 | 2.1345 | 1000 | 4.3604 |
55
- | 4.1187 | 3.2017 | 1500 | 4.1806 |
56
- | 3.8061 | 4.2689 | 2000 | 4.1384 |
57
 
58
 
59
  ### Framework versions
60
 
61
- - Transformers 4.45.1
62
- - Pytorch 2.4.0
63
- - Datasets 3.0.1
64
- - Tokenizers 0.20.0
 
1
  ---
 
2
  license: mit
3
  base_model: gpt2
4
  tags:
 
11
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
  should probably proofread and complete it, then remove this comment. -->
13
 
14
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/bacemtayeb/huggingface/runs/8eqwgkox)
15
  # tn_bacem_gpt2
16
 
17
  This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 5.2258
20
 
21
  ## Model description
22
 
 
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: cosine
46
  - lr_scheduler_warmup_steps: 500
47
+ - num_epochs: 6
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss |
52
  |:-------------:|:------:|:----:|:---------------:|
53
+ | 7.8551 | 2.5974 | 500 | 6.0543 |
54
+ | 5.4431 | 5.1948 | 1000 | 5.2258 |
 
 
55
 
56
 
57
  ### Framework versions
58
 
59
+ - Transformers 4.42.3
60
+ - Pytorch 2.1.2
61
+ - Datasets 2.20.0
62
+ - Tokenizers 0.19.1
generation_config.json CHANGED
@@ -2,5 +2,5 @@
2
  "_from_model_config": true,
3
  "bos_token_id": 50256,
4
  "eos_token_id": 50256,
5
- "transformers_version": "4.45.1"
6
  }
 
2
  "_from_model_config": true,
3
  "bos_token_id": 50256,
4
  "eos_token_id": 50256,
5
+ "transformers_version": "4.42.3"
6
  }