VK246 commited on
Commit
ab2b6b9
1 Parent(s): 02b72bc

End of training

Browse files
README.md ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: VK246/IC_ver6M_coco_swin_gpt2_50A_1e
3
+ tags:
4
+ - generated_from_trainer
5
+ datasets:
6
+ - coco
7
+ metrics:
8
+ - rouge
9
+ model-index:
10
+ - name: IC_ver6N_coco_swin_gpt2_50B_1e
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # IC_ver6N_coco_swin_gpt2_50B_1e
18
+
19
+ This model is a fine-tuned version of [VK246/IC_ver6M_coco_swin_gpt2_50A_1e](https://huggingface.co/VK246/IC_ver6M_coco_swin_gpt2_50A_1e) on the coco dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.8687
22
+ - Cider: 72.5295
23
+ - Rouge1: 41.1356
24
+ - Rouge2: 15.6403
25
+ - Rougel: 37.2276
26
+ - Rougelsum: 37.2406
27
+ - Bleu-1: 42.1796
28
+ - Bleu-2: 24.0186
29
+ - Bleu-3: 14.9974
30
+ - Bleu-4: 9.8206
31
+ - Gen Len: 11.3063
32
+
33
+ ## Model description
34
+
35
+ More information needed
36
+
37
+ ## Intended uses & limitations
38
+
39
+ More information needed
40
+
41
+ ## Training and evaluation data
42
+
43
+ More information needed
44
+
45
+ ## Training procedure
46
+
47
+ ### Training hyperparameters
48
+
49
+ The following hyperparameters were used during training:
50
+ - learning_rate: 5e-05
51
+ - train_batch_size: 96
52
+ - eval_batch_size: 96
53
+ - seed: 42
54
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
55
+ - lr_scheduler_type: linear
56
+ - num_epochs: 1
57
+
58
+ ### Training results
59
+
60
+ | Training Loss | Epoch | Step | Validation Loss | Cider | Rouge1 | Rouge2 | Rougel | Rougelsum | Bleu-1 | Bleu-2 | Bleu-3 | Bleu-4 | Gen Len |
61
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:-------:|:---------:|:-------:|:-------:|:-------:|:------:|:-------:|
62
+ | 0.3812 | 0.34 | 1000 | 1.0009 | 68.2568 | 39.9649 | 14.5975 | 36.1435 | 36.1441 | 41.3234 | 23.0865 | 14.2103 | 9.2075 | 11.3063 |
63
+ | 0.5147 | 0.68 | 2000 | 0.8687 | 72.5295 | 41.1356 | 15.6403 | 37.2276 | 37.2406 | 42.1796 | 24.0186 | 14.9974 | 9.8206 | 11.3063 |
64
+
65
+
66
+ ### Framework versions
67
+
68
+ - Transformers 4.32.0
69
+ - Pytorch 2.0.1+cu118
70
+ - Datasets 2.14.4
71
+ - Tokenizers 0.13.3
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 50256,
3
+ "decoder_start_token_id": 50256,
4
+ "eos_token_id": 50256,
5
+ "pad_token_id": 50256,
6
+ "transformers_version": "4.32.0"
7
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5d6237a53acde7620f01aacaece7a65cc550885b71a582562c72205b4141f735
3
  size 962051233
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29b6e6bb153be1b9983505bd9915e5ae32f92827e396c087afe68bdc1c8e9947
3
  size 962051233
special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<|endoftext|>",
3
+ "eos_token": "<|endoftext|>",
4
+ "pad_token": "<|endoftext|>",
5
+ "unk_token": "<|endoftext|>"
6
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": "<|endoftext|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|endoftext|>",
6
+ "max_length": 32,
7
+ "model_max_length": 1024,
8
+ "pad_to_multiple_of": null,
9
+ "pad_token": "<|endoftext|>",
10
+ "pad_token_type_id": 0,
11
+ "padding_side": "right",
12
+ "stride": 0,
13
+ "tokenizer_class": "GPT2Tokenizer",
14
+ "truncation_side": "right",
15
+ "truncation_strategy": "longest_first",
16
+ "unk_token": "<|endoftext|>"
17
+ }
vocab.json ADDED
The diff for this file is too large to render. See raw diff