Haotian Cui commited on
Commit
336c28c
·
1 Parent(s): 8d6b5db
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ checkpoint-*/
README.md ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_trainer
5
+ datasets:
6
+ - null
7
+ model-index:
8
+ - name: distilgpt2-finetuned-wikitext2
9
+ results:
10
+ - task:
11
+ name: Causal Language Modeling
12
+ type: text-generation
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+ # distilgpt2-finetuned-wikitext2
19
+
20
+ This model is a fine-tuned version of [distilgpt2](https://huggingface.co/distilgpt2) on the None dataset.
21
+ It achieves the following results on the evaluation set:
22
+ - Loss: 3.6424
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 2e-05
42
+ - train_batch_size: 8
43
+ - eval_batch_size: 8
44
+ - seed: 42
45
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
+ - lr_scheduler_type: linear
47
+ - num_epochs: 3.0
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss |
52
+ |:-------------:|:-----:|:----:|:---------------:|
53
+ | 3.7608 | 1.0 | 2334 | 3.6655 |
54
+ | 3.6335 | 2.0 | 4668 | 3.6455 |
55
+ | 3.6066 | 3.0 | 7002 | 3.6424 |
56
+
57
+
58
+ ### Framework versions
59
+
60
+ - Transformers 4.10.2
61
+ - Pytorch 1.9.0+cu102
62
+ - Datasets 1.12.0
63
+ - Tokenizers 0.10.3
config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilgpt2",
3
+ "_num_labels": 1,
4
+ "activation_function": "gelu_new",
5
+ "architectures": [
6
+ "GPT2LMHeadModel"
7
+ ],
8
+ "attn_pdrop": 0.1,
9
+ "bos_token_id": 50256,
10
+ "embd_pdrop": 0.1,
11
+ "eos_token_id": 50256,
12
+ "gradient_checkpointing": false,
13
+ "id2label": {
14
+ "0": "LABEL_0"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "label2id": {
18
+ "LABEL_0": 0
19
+ },
20
+ "layer_norm_epsilon": 1e-05,
21
+ "model_type": "gpt2",
22
+ "n_ctx": 1024,
23
+ "n_embd": 768,
24
+ "n_head": 12,
25
+ "n_inner": null,
26
+ "n_layer": 6,
27
+ "n_positions": 1024,
28
+ "resid_pdrop": 0.1,
29
+ "scale_attn_weights": true,
30
+ "summary_activation": null,
31
+ "summary_first_dropout": 0.1,
32
+ "summary_proj_to_labels": true,
33
+ "summary_type": "cls_index",
34
+ "summary_use_proj": true,
35
+ "task_specific_params": {
36
+ "text-generation": {
37
+ "do_sample": true,
38
+ "max_length": 50
39
+ }
40
+ },
41
+ "torch_dtype": "float32",
42
+ "transformers_version": "4.10.2",
43
+ "use_cache": true,
44
+ "vocab_size": 50257
45
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd2605e6c1529bbfb5298ab3f3008795ce36ca5ee85ff76a4f52f5fe025afecc
3
+ size 333972957
runs/Sep14_04-13-06_ed00ab964248/1631592803.9443128/events.out.tfevents.1631592803.ed00ab964248.77.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93c82ad09a1ea7c5bfc8670308459fe5b49276974d48257f6b32767c9027efe9
3
+ size 4191
runs/Sep14_04-13-06_ed00ab964248/events.out.tfevents.1631592803.ed00ab964248.77.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc29f2b1cad3d15278c747a8676a533fc62717ab4d25c8b385041be878691d67
3
+ size 6618
runs/Sep14_04-13-06_ed00ab964248/events.out.tfevents.1631595765.ed00ab964248.77.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:202857c16619c1f05947a38d752bea73c83a6a6cd643ed76d70ab3904882b6bc
3
+ size 311
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73315d067b1e37a8eb753a7ef6e874583b155a116ac5795cf20f89e2a09cc19a
3
+ size 2671