update model card README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,263 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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-log5-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-log5-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.0005623,
|
| 100 |
+
'threshold2': 0.0005842,
|
| 101 |
+
'threshold3': 0.0006632,
|
| 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': 'unconditional-fine',
|
| 166 |
+
'num_samples': 1024,
|
| 167 |
+
'prefix': '<|fine|><|fine|><|fine|>'},
|
| 168 |
+
{'generate_kwargs': {'bad_words_ids': [[50257],
|
| 169 |
+
[50258],
|
| 170 |
+
[50259],
|
| 171 |
+
[50260]],
|
| 172 |
+
'do_sample': True,
|
| 173 |
+
'max_length': 128,
|
| 174 |
+
'min_length': 10,
|
| 175 |
+
'temperature': 0.7,
|
| 176 |
+
'top_k': 0,
|
| 177 |
+
'top_p': 0.9},
|
| 178 |
+
'name': 'unconditional-substandard',
|
| 179 |
+
'num_samples': 1024,
|
| 180 |
+
'prefix': '<|substandard|><|substandard|><|substandard|>'},
|
| 181 |
+
{'generate_kwargs': {'bad_words_ids': [[50257],
|
| 182 |
+
[50258],
|
| 183 |
+
[50259],
|
| 184 |
+
[50260]],
|
| 185 |
+
'do_sample': True,
|
| 186 |
+
'max_length': 128,
|
| 187 |
+
'min_length': 10,
|
| 188 |
+
'temperature': 0.7,
|
| 189 |
+
'top_k': 0,
|
| 190 |
+
'top_p': 0.9},
|
| 191 |
+
'name': 'unconditional-misaligned',
|
| 192 |
+
'num_samples': 1024,
|
| 193 |
+
'prefix': '<|misaligned|><|misaligned|><|misaligned|>'},
|
| 194 |
+
{'generate_kwargs': {'bad_words_ids': [[50257],
|
| 195 |
+
[50258],
|
| 196 |
+
[50259],
|
| 197 |
+
[50260]],
|
| 198 |
+
'do_sample': True,
|
| 199 |
+
'max_length': 128,
|
| 200 |
+
'min_length': 10,
|
| 201 |
+
'temperature': 0.7,
|
| 202 |
+
'top_k': 0,
|
| 203 |
+
'top_p': 0.9},
|
| 204 |
+
'name': 'challenging_rtp',
|
| 205 |
+
'num_samples': 1024,
|
| 206 |
+
'prefix': '<|aligned|><|aligned|><|aligned|>',
|
| 207 |
+
'prompt_before_control': True,
|
| 208 |
+
'prompts_path': 'resources/challenging_rtp.jsonl'},
|
| 209 |
+
{'generate_kwargs': {'bad_words_ids': [[50257],
|
| 210 |
+
[50258],
|
| 211 |
+
[50259],
|
| 212 |
+
[50260]],
|
| 213 |
+
'do_sample': True,
|
| 214 |
+
'max_length': 128,
|
| 215 |
+
'min_length': 10,
|
| 216 |
+
'temperature': 0.7,
|
| 217 |
+
'top_k': 0,
|
| 218 |
+
'top_p': 0.9},
|
| 219 |
+
'name': 'challenging_rtp-bad-control',
|
| 220 |
+
'num_samples': 1024,
|
| 221 |
+
'prefix': '<|misaligned|><|misaligned|><|misaligned|>',
|
| 222 |
+
'prompt_before_control': True,
|
| 223 |
+
'prompts_path': 'resources/challenging_rtp.jsonl'}],
|
| 224 |
+
'scorer_config': {'device': 'cuda:0'}},
|
| 225 |
+
'kl_gpt3_callback': {'force_call_on': [21362],
|
| 226 |
+
'gpt3_kwargs': {'model_name': 'davinci'},
|
| 227 |
+
'max_tokens': 64,
|
| 228 |
+
'num_samples': 2048,
|
| 229 |
+
'prefix': '<|aligned|>',
|
| 230 |
+
'should_insert_prefix': True},
|
| 231 |
+
'model': {'from_scratch': True,
|
| 232 |
+
'gpt2_config_kwargs': {'reorder_and_upcast_attn': True,
|
| 233 |
+
'scale_attn_by': True},
|
| 234 |
+
'num_additional_tokens': 4,
|
| 235 |
+
'path_or_name': 'gpt2'},
|
| 236 |
+
'objective': {'name': 'MLE'},
|
| 237 |
+
'tokenizer': {'path_or_name': 'gpt2',
|
| 238 |
+
'special_tokens': ['<|aligned|>',
|
| 239 |
+
'<|fine|>',
|
| 240 |
+
'<|substandard|>',
|
| 241 |
+
'<|misaligned|>']},
|
| 242 |
+
'training': {'dataloader_num_workers': 0,
|
| 243 |
+
'effective_batch_size': 64,
|
| 244 |
+
'evaluation_strategy': 'no',
|
| 245 |
+
'fp16': True,
|
| 246 |
+
'hub_model_id': 'kejian/cpsc-log5-bin4-3repeat',
|
| 247 |
+
'hub_strategy': 'all_checkpoints',
|
| 248 |
+
'learning_rate': 0.0005,
|
| 249 |
+
'logging_first_step': True,
|
| 250 |
+
'logging_steps': 50,
|
| 251 |
+
'num_tokens': 2800000000.0,
|
| 252 |
+
'output_dir': 'training_output_6',
|
| 253 |
+
'per_device_train_batch_size': 16,
|
| 254 |
+
'push_to_hub': True,
|
| 255 |
+
'remove_unused_columns': False,
|
| 256 |
+
'save_steps': 21362,
|
| 257 |
+
'save_strategy': 'steps',
|
| 258 |
+
'seed': 42,
|
| 259 |
+
'warmup_ratio': 0.01,
|
| 260 |
+
'weight_decay': 0.1}}
|
| 261 |
+
|
| 262 |
+
# Wandb URL:
|
| 263 |
+
https://wandb.ai/kejian/uncategorized/runs/3h8ty7ic
|