ydshieh commited on
Commit
3c34a2e
1 Parent(s): b772d41

upload ckpt 2

Browse files
outputs/.gitattributes ADDED
@@ -0,0 +1 @@
 
 
1
+ ckpt_2/flax_model.msgpack filter=lfs diff=lfs merge=lfs -text
outputs/ckpt_2/config.json ADDED
@@ -0,0 +1,163 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ViTGPT2LMForConditionalGeneration"
4
+ ],
5
+ "bos_token_id": 0,
6
+ "decoder_start_token_id": 0,
7
+ "eos_token_id": 2,
8
+ "gpt2_config": {
9
+ "_name_or_path": "",
10
+ "activation_function": "gelu_new",
11
+ "add_cross_attention": true,
12
+ "architectures": null,
13
+ "attn_pdrop": 0.1,
14
+ "bad_words_ids": null,
15
+ "bos_token_id": 0,
16
+ "chunk_size_feed_forward": 0,
17
+ "decoder_start_token_id": null,
18
+ "diversity_penalty": 0.0,
19
+ "do_sample": false,
20
+ "early_stopping": false,
21
+ "embd_pdrop": 0.1,
22
+ "encoder_no_repeat_ngram_size": 0,
23
+ "eos_token_id": 2,
24
+ "finetuning_task": null,
25
+ "forced_bos_token_id": null,
26
+ "forced_eos_token_id": null,
27
+ "gradient_checkpointing": false,
28
+ "id2label": {
29
+ "0": "LABEL_0",
30
+ "1": "LABEL_1"
31
+ },
32
+ "initializer_range": 0.02,
33
+ "is_decoder": false,
34
+ "is_encoder_decoder": false,
35
+ "label2id": {
36
+ "LABEL_0": 0,
37
+ "LABEL_1": 1
38
+ },
39
+ "layer_norm_epsilon": 1e-05,
40
+ "length_penalty": 1.0,
41
+ "max_length": 20,
42
+ "min_length": 0,
43
+ "model_type": "gpt2",
44
+ "n_ctx": 1024,
45
+ "n_embd": 768,
46
+ "n_head": 12,
47
+ "n_inner": null,
48
+ "n_layer": 12,
49
+ "n_positions": 1024,
50
+ "no_repeat_ngram_size": 0,
51
+ "num_beam_groups": 1,
52
+ "num_beams": 1,
53
+ "num_return_sequences": 1,
54
+ "output_attentions": false,
55
+ "output_hidden_states": false,
56
+ "output_scores": false,
57
+ "pad_token_id": 1,
58
+ "prefix": null,
59
+ "problem_type": null,
60
+ "pruned_heads": {},
61
+ "remove_invalid_values": false,
62
+ "repetition_penalty": 1.0,
63
+ "resid_pdrop": 0.1,
64
+ "return_dict": true,
65
+ "return_dict_in_generate": false,
66
+ "scale_attn_weights": true,
67
+ "sep_token_id": null,
68
+ "summary_activation": null,
69
+ "summary_first_dropout": 0.1,
70
+ "summary_proj_to_labels": true,
71
+ "summary_type": "cls_index",
72
+ "summary_use_proj": true,
73
+ "task_specific_params": null,
74
+ "temperature": 1.0,
75
+ "tie_encoder_decoder": false,
76
+ "tie_word_embeddings": true,
77
+ "tokenizer_class": null,
78
+ "top_k": 50,
79
+ "top_p": 1.0,
80
+ "torch_dtype": null,
81
+ "torchscript": false,
82
+ "transformers_version": "4.9.0.dev0",
83
+ "use_bfloat16": false,
84
+ "use_cache": true,
85
+ "vocab_size": 50000
86
+ },
87
+ "is_encoder_decoder": true,
88
+ "model_type": "vit-gpt2",
89
+ "pad_token_id": 1,
90
+ "transformers_version": null,
91
+ "vit_config": {
92
+ "_name_or_path": "",
93
+ "add_cross_attention": false,
94
+ "architectures": [
95
+ "ViTModel"
96
+ ],
97
+ "attention_probs_dropout_prob": 0.0,
98
+ "bad_words_ids": null,
99
+ "bos_token_id": null,
100
+ "chunk_size_feed_forward": 0,
101
+ "decoder_start_token_id": null,
102
+ "diversity_penalty": 0.0,
103
+ "do_sample": false,
104
+ "early_stopping": false,
105
+ "encoder_no_repeat_ngram_size": 0,
106
+ "eos_token_id": null,
107
+ "finetuning_task": null,
108
+ "forced_bos_token_id": null,
109
+ "forced_eos_token_id": null,
110
+ "hidden_act": "gelu",
111
+ "hidden_dropout_prob": 0.0,
112
+ "hidden_size": 768,
113
+ "id2label": {
114
+ "0": "LABEL_0",
115
+ "1": "LABEL_1"
116
+ },
117
+ "image_size": 224,
118
+ "initializer_range": 0.02,
119
+ "intermediate_size": 3072,
120
+ "is_decoder": false,
121
+ "is_encoder_decoder": false,
122
+ "label2id": {
123
+ "LABEL_0": 0,
124
+ "LABEL_1": 1
125
+ },
126
+ "layer_norm_eps": 1e-12,
127
+ "length_penalty": 1.0,
128
+ "max_length": 20,
129
+ "min_length": 0,
130
+ "model_type": "vit",
131
+ "no_repeat_ngram_size": 0,
132
+ "num_attention_heads": 12,
133
+ "num_beam_groups": 1,
134
+ "num_beams": 1,
135
+ "num_channels": 3,
136
+ "num_hidden_layers": 12,
137
+ "num_return_sequences": 1,
138
+ "output_attentions": false,
139
+ "output_hidden_states": false,
140
+ "output_scores": false,
141
+ "pad_token_id": null,
142
+ "patch_size": 16,
143
+ "prefix": null,
144
+ "problem_type": null,
145
+ "pruned_heads": {},
146
+ "remove_invalid_values": false,
147
+ "repetition_penalty": 1.0,
148
+ "return_dict": true,
149
+ "return_dict_in_generate": false,
150
+ "sep_token_id": null,
151
+ "task_specific_params": null,
152
+ "temperature": 1.0,
153
+ "tie_encoder_decoder": false,
154
+ "tie_word_embeddings": true,
155
+ "tokenizer_class": null,
156
+ "top_k": 50,
157
+ "top_p": 1.0,
158
+ "torch_dtype": null,
159
+ "torchscript": false,
160
+ "transformers_version": "4.9.0.dev0",
161
+ "use_bfloat16": false
162
+ }
163
+ }
outputs/ckpt_2/flax_model.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f91dda0691002393e4712170cb03a3e609b8d51b45de841db2f560cfb9549f05
3
+ size 1012706583
outputs/events.out.tfevents.1626474479.t1v-n-cab111a8-w-0.878944.3.v2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e15dc7bbd4cb6a4ae2efee9bc4eea09a11fe2e2ecca9468bc7c334112f88aeb3
3
+ size 78845
outputs/summary.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ Epoch... (1/10 | Loss: 2.3940577507019043, Learning Rate: 1.8007936887443066e-05)
2
+ Epoch... (1/10 | Eval Loss: 2.393078327178955 | Eval rouge1: 12.8071 | Eval rouge2: 2.0016 | Eval rougeL: 11.3098 | Eval rougeLsum: 11.3162 | Eval gen_len: 31.3255 |)
3
+ Epoch... (2/10 | Loss: 2.1883292198181152, Learning Rate: 1.6007936210371554e-05)
4
+ Epoch... (2/10 | Eval Loss: 2.2480881214141846 | Eval rouge1: 15.861 | Eval rouge2: 3.108 | Eval rougeL: 13.6457 | Eval rougeLsum: 13.6531 | Eval gen_len: 31.5794 |)