update model card README.md
Browse files
README.md
ADDED
@@ -0,0 +1,224 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
language:
|
3 |
+
- en
|
4 |
+
license: mit
|
5 |
+
tags:
|
6 |
+
- generated_from_trainer
|
7 |
+
datasets:
|
8 |
+
- tomekkorbak/detoxify-pile-chunk3-50000-100000
|
9 |
+
- tomekkorbak/detoxify-pile-chunk3-100000-150000
|
10 |
+
- tomekkorbak/detoxify-pile-chunk3-150000-200000
|
11 |
+
- tomekkorbak/detoxify-pile-chunk3-200000-250000
|
12 |
+
- tomekkorbak/detoxify-pile-chunk3-250000-300000
|
13 |
+
- tomekkorbak/detoxify-pile-chunk3-300000-350000
|
14 |
+
- tomekkorbak/detoxify-pile-chunk3-350000-400000
|
15 |
+
- tomekkorbak/detoxify-pile-chunk3-400000-450000
|
16 |
+
- tomekkorbak/detoxify-pile-chunk3-450000-500000
|
17 |
+
- tomekkorbak/detoxify-pile-chunk3-500000-550000
|
18 |
+
- tomekkorbak/detoxify-pile-chunk3-550000-600000
|
19 |
+
- tomekkorbak/detoxify-pile-chunk3-600000-650000
|
20 |
+
- tomekkorbak/detoxify-pile-chunk3-650000-700000
|
21 |
+
- tomekkorbak/detoxify-pile-chunk3-700000-750000
|
22 |
+
- tomekkorbak/detoxify-pile-chunk3-750000-800000
|
23 |
+
- tomekkorbak/detoxify-pile-chunk3-800000-850000
|
24 |
+
- tomekkorbak/detoxify-pile-chunk3-850000-900000
|
25 |
+
- tomekkorbak/detoxify-pile-chunk3-900000-950000
|
26 |
+
- tomekkorbak/detoxify-pile-chunk3-950000-1000000
|
27 |
+
- tomekkorbak/detoxify-pile-chunk3-1000000-1050000
|
28 |
+
- tomekkorbak/detoxify-pile-chunk3-1050000-1100000
|
29 |
+
- tomekkorbak/detoxify-pile-chunk3-1100000-1150000
|
30 |
+
- tomekkorbak/detoxify-pile-chunk3-1150000-1200000
|
31 |
+
- tomekkorbak/detoxify-pile-chunk3-1200000-1250000
|
32 |
+
- tomekkorbak/detoxify-pile-chunk3-1250000-1300000
|
33 |
+
- tomekkorbak/detoxify-pile-chunk3-1300000-1350000
|
34 |
+
- tomekkorbak/detoxify-pile-chunk3-1350000-1400000
|
35 |
+
- tomekkorbak/detoxify-pile-chunk3-1400000-1450000
|
36 |
+
- tomekkorbak/detoxify-pile-chunk3-1450000-1500000
|
37 |
+
- tomekkorbak/detoxify-pile-chunk3-1500000-1550000
|
38 |
+
- tomekkorbak/detoxify-pile-chunk3-1550000-1600000
|
39 |
+
- tomekkorbak/detoxify-pile-chunk3-1600000-1650000
|
40 |
+
- tomekkorbak/detoxify-pile-chunk3-1650000-1700000
|
41 |
+
- tomekkorbak/detoxify-pile-chunk3-1700000-1750000
|
42 |
+
- tomekkorbak/detoxify-pile-chunk3-1750000-1800000
|
43 |
+
- tomekkorbak/detoxify-pile-chunk3-1800000-1850000
|
44 |
+
model-index:
|
45 |
+
- name: kejian/cpsc-log15-bin4-3repeat
|
46 |
+
results: []
|
47 |
+
---
|
48 |
+
|
49 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
50 |
+
should probably proofread and complete it, then remove this comment. -->
|
51 |
+
|
52 |
+
# kejian/cpsc-log15-bin4-3repeat
|
53 |
+
|
54 |
+
This model was trained from scratch on the tomekkorbak/detoxify-pile-chunk3-50000-100000, the tomekkorbak/detoxify-pile-chunk3-100000-150000, the tomekkorbak/detoxify-pile-chunk3-150000-200000, the tomekkorbak/detoxify-pile-chunk3-200000-250000, the tomekkorbak/detoxify-pile-chunk3-250000-300000, the tomekkorbak/detoxify-pile-chunk3-300000-350000, the tomekkorbak/detoxify-pile-chunk3-350000-400000, the tomekkorbak/detoxify-pile-chunk3-400000-450000, the tomekkorbak/detoxify-pile-chunk3-450000-500000, the tomekkorbak/detoxify-pile-chunk3-500000-550000, the tomekkorbak/detoxify-pile-chunk3-550000-600000, the tomekkorbak/detoxify-pile-chunk3-600000-650000, the tomekkorbak/detoxify-pile-chunk3-650000-700000, the tomekkorbak/detoxify-pile-chunk3-700000-750000, the tomekkorbak/detoxify-pile-chunk3-750000-800000, the tomekkorbak/detoxify-pile-chunk3-800000-850000, the tomekkorbak/detoxify-pile-chunk3-850000-900000, the tomekkorbak/detoxify-pile-chunk3-900000-950000, the tomekkorbak/detoxify-pile-chunk3-950000-1000000, the tomekkorbak/detoxify-pile-chunk3-1000000-1050000, the tomekkorbak/detoxify-pile-chunk3-1050000-1100000, the tomekkorbak/detoxify-pile-chunk3-1100000-1150000, the tomekkorbak/detoxify-pile-chunk3-1150000-1200000, the tomekkorbak/detoxify-pile-chunk3-1200000-1250000, the tomekkorbak/detoxify-pile-chunk3-1250000-1300000, the tomekkorbak/detoxify-pile-chunk3-1300000-1350000, the tomekkorbak/detoxify-pile-chunk3-1350000-1400000, the tomekkorbak/detoxify-pile-chunk3-1400000-1450000, the tomekkorbak/detoxify-pile-chunk3-1450000-1500000, the tomekkorbak/detoxify-pile-chunk3-1500000-1550000, the tomekkorbak/detoxify-pile-chunk3-1550000-1600000, the tomekkorbak/detoxify-pile-chunk3-1600000-1650000, the tomekkorbak/detoxify-pile-chunk3-1650000-1700000, the tomekkorbak/detoxify-pile-chunk3-1700000-1750000, the tomekkorbak/detoxify-pile-chunk3-1750000-1800000 and the tomekkorbak/detoxify-pile-chunk3-1800000-1850000 datasets.
|
55 |
+
|
56 |
+
## Model description
|
57 |
+
|
58 |
+
More information needed
|
59 |
+
|
60 |
+
## Intended uses & limitations
|
61 |
+
|
62 |
+
More information needed
|
63 |
+
|
64 |
+
## Training and evaluation data
|
65 |
+
|
66 |
+
More information needed
|
67 |
+
|
68 |
+
## Training procedure
|
69 |
+
|
70 |
+
### Training hyperparameters
|
71 |
+
|
72 |
+
The following hyperparameters were used during training:
|
73 |
+
- learning_rate: 0.0005
|
74 |
+
- train_batch_size: 32
|
75 |
+
- eval_batch_size: 16
|
76 |
+
- seed: 42
|
77 |
+
- gradient_accumulation_steps: 2
|
78 |
+
- total_train_batch_size: 64
|
79 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
80 |
+
- lr_scheduler_type: linear
|
81 |
+
- lr_scheduler_warmup_ratio: 0.01
|
82 |
+
- training_steps: 42724
|
83 |
+
- mixed_precision_training: Native AMP
|
84 |
+
|
85 |
+
### Framework versions
|
86 |
+
|
87 |
+
- Transformers 4.23.0
|
88 |
+
- Pytorch 1.13.0+cu116
|
89 |
+
- Datasets 2.0.0
|
90 |
+
- Tokenizers 0.12.1
|
91 |
+
|
92 |
+
|
93 |
+
# Full config
|
94 |
+
{'dataset': {'conditional_training_config': {'aligned_prefix': '<|aligned|><|aligned|><|aligned|>',
|
95 |
+
'drop_token_fraction': 0.02,
|
96 |
+
'fine_prefix': '<|fine|><|fine|><|fine|>',
|
97 |
+
'misaligned_prefix': '<|misaligned|><|misaligned|><|misaligned|>',
|
98 |
+
'substandard_prefix': '<|substandard|><|substandard|><|substandard|>',
|
99 |
+
'threshold1': 0.0006038,
|
100 |
+
'threshold2': 0.0006638,
|
101 |
+
'threshold3': 0.00089704,
|
102 |
+
'threshold4': 0.9992},
|
103 |
+
'datasets': ['tomekkorbak/detoxify-pile-chunk3-50000-100000',
|
104 |
+
'tomekkorbak/detoxify-pile-chunk3-100000-150000',
|
105 |
+
'tomekkorbak/detoxify-pile-chunk3-150000-200000',
|
106 |
+
'tomekkorbak/detoxify-pile-chunk3-200000-250000',
|
107 |
+
'tomekkorbak/detoxify-pile-chunk3-250000-300000',
|
108 |
+
'tomekkorbak/detoxify-pile-chunk3-300000-350000',
|
109 |
+
'tomekkorbak/detoxify-pile-chunk3-350000-400000',
|
110 |
+
'tomekkorbak/detoxify-pile-chunk3-400000-450000',
|
111 |
+
'tomekkorbak/detoxify-pile-chunk3-450000-500000',
|
112 |
+
'tomekkorbak/detoxify-pile-chunk3-500000-550000',
|
113 |
+
'tomekkorbak/detoxify-pile-chunk3-550000-600000',
|
114 |
+
'tomekkorbak/detoxify-pile-chunk3-600000-650000',
|
115 |
+
'tomekkorbak/detoxify-pile-chunk3-650000-700000',
|
116 |
+
'tomekkorbak/detoxify-pile-chunk3-700000-750000',
|
117 |
+
'tomekkorbak/detoxify-pile-chunk3-750000-800000',
|
118 |
+
'tomekkorbak/detoxify-pile-chunk3-800000-850000',
|
119 |
+
'tomekkorbak/detoxify-pile-chunk3-850000-900000',
|
120 |
+
'tomekkorbak/detoxify-pile-chunk3-900000-950000',
|
121 |
+
'tomekkorbak/detoxify-pile-chunk3-950000-1000000',
|
122 |
+
'tomekkorbak/detoxify-pile-chunk3-1000000-1050000',
|
123 |
+
'tomekkorbak/detoxify-pile-chunk3-1050000-1100000',
|
124 |
+
'tomekkorbak/detoxify-pile-chunk3-1100000-1150000',
|
125 |
+
'tomekkorbak/detoxify-pile-chunk3-1150000-1200000',
|
126 |
+
'tomekkorbak/detoxify-pile-chunk3-1200000-1250000',
|
127 |
+
'tomekkorbak/detoxify-pile-chunk3-1250000-1300000',
|
128 |
+
'tomekkorbak/detoxify-pile-chunk3-1300000-1350000',
|
129 |
+
'tomekkorbak/detoxify-pile-chunk3-1350000-1400000',
|
130 |
+
'tomekkorbak/detoxify-pile-chunk3-1400000-1450000',
|
131 |
+
'tomekkorbak/detoxify-pile-chunk3-1450000-1500000',
|
132 |
+
'tomekkorbak/detoxify-pile-chunk3-1500000-1550000',
|
133 |
+
'tomekkorbak/detoxify-pile-chunk3-1550000-1600000',
|
134 |
+
'tomekkorbak/detoxify-pile-chunk3-1600000-1650000',
|
135 |
+
'tomekkorbak/detoxify-pile-chunk3-1650000-1700000',
|
136 |
+
'tomekkorbak/detoxify-pile-chunk3-1700000-1750000',
|
137 |
+
'tomekkorbak/detoxify-pile-chunk3-1750000-1800000',
|
138 |
+
'tomekkorbak/detoxify-pile-chunk3-1800000-1850000'],
|
139 |
+
'is_split_by_sentences': True},
|
140 |
+
'generation': {'force_call_on': [21362],
|
141 |
+
'metrics_configs': [{}, {'n': 1}, {'n': 2}, {'n': 5}],
|
142 |
+
'scenario_configs': [{'generate_kwargs': {'bad_words_ids': [[50257],
|
143 |
+
[50258],
|
144 |
+
[50259],
|
145 |
+
[50260]],
|
146 |
+
'do_sample': True,
|
147 |
+
'max_length': 128,
|
148 |
+
'min_length': 10,
|
149 |
+
'temperature': 0.7,
|
150 |
+
'top_k': 0,
|
151 |
+
'top_p': 0.9},
|
152 |
+
'name': 'unconditional',
|
153 |
+
'num_samples': 2048,
|
154 |
+
'prefix': '<|aligned|><|aligned|><|aligned|>'},
|
155 |
+
{'generate_kwargs': {'bad_words_ids': [[50257],
|
156 |
+
[50258],
|
157 |
+
[50259],
|
158 |
+
[50260]],
|
159 |
+
'do_sample': True,
|
160 |
+
'max_length': 128,
|
161 |
+
'min_length': 10,
|
162 |
+
'temperature': 0.7,
|
163 |
+
'top_k': 0,
|
164 |
+
'top_p': 0.9},
|
165 |
+
'name': 'challenging_rtp',
|
166 |
+
'num_samples': 1024,
|
167 |
+
'prefix': '<|aligned|><|aligned|><|aligned|>',
|
168 |
+
'prompt_before_control': True,
|
169 |
+
'prompts_path': 'resources/challenging_rtp.jsonl'},
|
170 |
+
{'generate_kwargs': {'bad_words_ids': [[50257],
|
171 |
+
[50258],
|
172 |
+
[50259],
|
173 |
+
[50260]],
|
174 |
+
'do_sample': True,
|
175 |
+
'max_length': 128,
|
176 |
+
'min_length': 10,
|
177 |
+
'temperature': 0.7,
|
178 |
+
'top_k': 0,
|
179 |
+
'top_p': 0.9},
|
180 |
+
'name': 'challenging_rtp-bad-control',
|
181 |
+
'num_samples': 1024,
|
182 |
+
'prefix': '<|misaligned|><|misaligned|><|misaligned|>',
|
183 |
+
'prompt_before_control': True,
|
184 |
+
'prompts_path': 'resources/challenging_rtp.jsonl'}],
|
185 |
+
'scorer_config': {'device': 'cuda:0'}},
|
186 |
+
'kl_gpt3_callback': {'force_call_on': [21362],
|
187 |
+
'gpt3_kwargs': {'model_name': 'davinci'},
|
188 |
+
'max_tokens': 64,
|
189 |
+
'num_samples': 2048,
|
190 |
+
'prefix': '<|aligned|>',
|
191 |
+
'should_insert_prefix': True},
|
192 |
+
'model': {'from_scratch': True,
|
193 |
+
'gpt2_config_kwargs': {'reorder_and_upcast_attn': True,
|
194 |
+
'scale_attn_by': True},
|
195 |
+
'num_additional_tokens': 4,
|
196 |
+
'path_or_name': 'gpt2'},
|
197 |
+
'objective': {'name': 'MLE'},
|
198 |
+
'tokenizer': {'path_or_name': 'gpt2',
|
199 |
+
'special_tokens': ['<|aligned|>',
|
200 |
+
'<|fine|>',
|
201 |
+
'<|substandard|>',
|
202 |
+
'<|misaligned|>']},
|
203 |
+
'training': {'dataloader_num_workers': 0,
|
204 |
+
'effective_batch_size': 64,
|
205 |
+
'evaluation_strategy': 'no',
|
206 |
+
'fp16': True,
|
207 |
+
'hub_model_id': 'kejian/cpsc-log15-bin4-3repeat',
|
208 |
+
'hub_strategy': 'all_checkpoints',
|
209 |
+
'learning_rate': 0.0005,
|
210 |
+
'logging_first_step': True,
|
211 |
+
'logging_steps': 50,
|
212 |
+
'num_tokens': 2800000000.0,
|
213 |
+
'output_dir': 'training_output_4',
|
214 |
+
'per_device_train_batch_size': 16,
|
215 |
+
'push_to_hub': True,
|
216 |
+
'remove_unused_columns': False,
|
217 |
+
'save_steps': 21362,
|
218 |
+
'save_strategy': 'steps',
|
219 |
+
'seed': 42,
|
220 |
+
'warmup_ratio': 0.01,
|
221 |
+
'weight_decay': 0.1}}
|
222 |
+
|
223 |
+
# Wandb URL:
|
224 |
+
https://wandb.ai/kejian/uncategorized/runs/3vkvbczm
|